Update app.py

Key Changes:

create_model_repo (Sanitization):

Sanitize model_name: Added model_name = re.sub(r"[^a-zA-Z0-9._-]", "-", model_name) to sanitize the model_name input. This replaces any invalid characters with hyphens.

Sanitize orgs_name: Added orgs_name = re.sub(r"[^a-zA-Z0-9._-]", "-", orgs_name) to sanitize the orgs_name input.

Sanitize in Main: Added sanitization.

main Function (Whitespace Stripping): The code to strip whitespace from all input strings is still present and correct.

Testing (Crucial):

Test with Invalid Characters: Try entering model names and organization names that contain spaces, special symbols (e.g., !@#$%^&*()_+=-[]\{}|;':",./<>?), and other characters that are not allowed in Hugging Face repository IDs. The code should now replace these with hyphens and create a valid repository.

Test with Valid Characters: Also test with valid model and organization names to ensure everything still works correctly.

Test with Empty Fields: Test with and without providing an organization name and model name.

Test with invalid token.

Test with URL, filepath, and repo name.

app.py

@@ -99,7 +99,7 @@ def download_model(model_path_or_url):
 
 
 def create_model_repo(api, user, orgs_name, model_name, make_private=False):
-    """Creates a Hugging Face model repository, handling missing inputs and sanitizing the username."""
+    """Creates a Hugging Face model repository, sanitizing inputs."""
 
     print("---- create_model_repo Called ----")
     print(f"  user: {user}")
@@ -110,18 +110,23 @@ def create_model_repo(api, user, orgs_name, model_name, make_private=False):
         model_name = f"converted-model-{datetime.now().strftime('%Y%m%d%H%M%S')}"
         print(f"  Using default model_name: {model_name}")
 
+    # --- Sanitize model_name and orgs_name ---
+    if orgs_name:
+        orgs_name = re.sub(r"[^a-zA-Z0-9._-]", "-", orgs_name)
+        print(f"  Sanitized orgs_name: {orgs_name}")
+    if model_name:
+        model_name = re.sub(r"[^a-zA-Z0-9._-]", "-", model_name)
+        print(f"  Sanitized model_name: {model_name}")
+
+
     if orgs_name:
         repo_id = f"{orgs_name}/{model_name.strip()}"
     elif user:
-        # --- Sanitize the username ---
-        sanitized_username = re.sub(r"[^a-zA-Z0-9._-]", "-", user['name'])  # Replace invalid chars with hyphens
-        print(f"  Original Username: {user['name']}") #Debugging
-        print(f"  Sanitized Username: {sanitized_username}") #Debugging
+        sanitized_username = re.sub(r"[^a-zA-Z0-9._-]", "-", user['name'])
+        print(f"  Sanitized Username: {sanitized_username}")
         repo_id = f"{sanitized_username}/{model_name.strip()}"
     else:
-        raise ValueError(
-            "Must provide either an organization name or be logged in."
-        )
+        raise ValueError("Must provide either an organization name or be logged in.")
 
     print(f"  repo_id: {repo_id}")
 
@@ -246,13 +251,19 @@ def main(
         print(f"---- Main Function Error: {error_message} ----")
         return error_message
 
-    # --- Strip Whitespace from Inputs ---
+    # --- Strip Whitespace and Sanitize from Inputs ---
     model_to_load = model_to_load.strip()
     reference_model = reference_model.strip()
     output_path = output_path.strip()
-    hf_token = hf_token.strip()  # Even though it's a password field, good practice
-    orgs_name = orgs_name.strip() if orgs_name else "" #Handle empty
-    model_name = model_name.strip() if model_name else "" #Handle empty
+    hf_token = hf_token.strip()  # Even though it's a password field
+    orgs_name = orgs_name.strip() if orgs_name else ""
+    model_name = model_name.strip() if model_name else ""
+
+    # --- Sanitize model_name and orgs_name ---
+    if orgs_name:
+        orgs_name = re.sub(r"[^a-zA-Z0-9._-]", "-", orgs_name)
+    if model_name:
+        model_name = re.sub(r"[^a-zA-Z0-9._-]", "-", model_name)
 
     try:
         convert_and_save_sdxl_to_diffusers(model_to_load, output_path, reference_model)
@@ -263,11 +274,12 @@ def main(
             print(f"Using default model_name: {model_name}")
 
         if orgs_name:
-            repo_id = f"{orgs_name}/{model_name.strip()}"
+            repo_id = f"{orgs_name}/{model_name}"
         elif user:
             # Sanitize username here as well:
             sanitized_username = re.sub(r"[^a-zA-Z0-9._-]", "-", user['name'])
-            repo_id = f"{sanitized_username}/{model_name.strip()}"
+            print(f"  Sanitized Username: {sanitized_username}")
+            repo_id = f"{sanitized_username}/{model_name}"
 
         else:  # Should never happen because of login, but good practice
             raise ValueError("Must provide either an organization name or be logged in.")
@@ -317,18 +329,15 @@ with gr.Blocks(css=css) as demo:
     - Direct URLs to model files
     - Hugging Face model repositories (e.g., 'my-org/my-model' or 'my-org/my-model/file.safetensors')
 
-    ### How To Use 
-    - Insert URL or Repository Information into Field 1 SDXL Checkpoint (Path, URL, or HF Repo)
-    - Optional: Insert Reference Diffusers Model (Optional)
-    - Optional: Output Path (Diffusers Format)
-    - Insert your HF Token WRITE:  Get your HF token here: [https://huggingface.co/settings/tokens](https://huggingface.co/settings/tokens)
-    - Organization Name (Optional)
-    - Create a Model Name for Output Purposes
-
     ### ℹ️ Important Notes:
     - This tool runs on **CPU**, conversion might be slower than on GPU.
     - For Hugging Face uploads, you need a **WRITE** token (not a read token).
+    - Get your HF token here: [https://huggingface.co/settings/tokens](https://huggingface.co/settings/tokens)
+
+    ### 💾 Memory Usage:
     - This space is configured for **FP16** precision to reduce memory usage.
+    - Close other applications during conversion.
+    - For large models, ensure you have at least 16GB of RAM.
 
     ### 💻 Source Code:
     - [GitHub Repository](https://github.com/Ktiseos-Nyx/Gradio-SDXL-Diffusers)
@@ -341,11 +350,11 @@ with gr.Blocks(css=css) as demo:
     with gr.Row():
         with gr.Column():
             model_to_load = gr.Textbox(
-                label="SDXL Checkpoint (Path, URL, or HF Repo)",  # Corrected Label
+                label="SDXL Checkpoint (Path, URL, or HF Repo)",
                 placeholder="Path, URL, or Hugging Face Repo ID (e.g., my-org/my-model or my-org/my-model/file.safetensors)",
             )
             reference_model = gr.Textbox(
-                label="Reference Diffusers Model (Optional)",  # Corrected Label
+                label="Reference Diffusers Model (Optional)",
                 placeholder="e.g., stabilityai/stable-diffusion-xl-base-1.0 (Leave blank for default)",
             )
             output_path = gr.Textbox(label="Output Path (Diffusers Format)", value="output")