Update app.py

app.py

@@ -25,11 +25,15 @@ def generate_files(title="Text Generation Tool", tool_description="This is a too
     ---
     '''.format(title)
 
+	tool_name = title.replace(" ","_").lower()
+	tool_class = title.replace(" ","")
+	tool_repo_id  = title.replace(" ","-")
+	
     # Generate tool config JSON content
     tool_config = {
         "description": tool_description,
-        "name": title.replace(" ","_").lower(),
-        "tool_class": "{}Tool".format(title.replace(" ",""))
+        "name": tool_name,
+        "tool_class": "{}Tool".format(tool_class)
     }
     tool_config_json = json.dumps(tool_config, indent=4)
 
@@ -39,7 +43,7 @@ def generate_files(title="Text Generation Tool", tool_description="This is a too
     import launch_gradio_demo
     from {} import {}
     launch_gradio_demo({}Tool)
-    '''.format( title.replace(" ","_").lower(), title.replace(" ",""), title.replace(" ",""))
+    '''.format( tool_name, tool_class, tool_class)
 
     # Generate requirements.txt content
     requirements_content = '''transformers>=4.29.0
@@ -64,45 +68,45 @@ class {}(Tool):
         generated_text = text_generator(prompt, max_length=500, num_return_sequences=1, temperature=0.7)
         print(generated_text)
         return generated_text
-'''.format(title.replace(" ",""), title.replace(" ","_").lower(), tool_description)
+'''.format(tool_class, tool_name, tool_description)
 
     # Create a new folder for the tool
-    os.makedirs(title.replace(" ",""), exist_ok=True)
+    os.makedirs(tool_class, exist_ok=True)
 
     # Write content to files
-    with open(f"{title.replace(" ","")}/README.md", "w") as readme_file:
+    with open(f"{tool_class}/README.md", "w") as readme_file:
         readme_file.write(readme_content)
 
-    with open(f"{title.replace(" ","")}/tool_config.json", "w") as tool_config_file:
+    with open(f"{tool_class}/tool_config.json", "w") as tool_config_file:
         tool_config_file.write(tool_config_json)
 
-    with open(f"{title.replace(" ","")}/app.py", "w") as app_py_file:
+    with open(f"{tool_class}/app.py", "w") as app_py_file:
         app_py_file.write(app_py_content)
 
-    with open(f"{title.replace(" ","")}/requirements.txt", "w") as requirements_file:
+    with open(f"{tool_class}/requirements.txt", "w") as requirements_file:
         requirements_file.write(requirements_content)
 
-    with open(f"{title.replace(" ","")}/app.py", "w") as text_generator_py_file:
+    with open(f"{tool_class}/app.py", "w") as text_generator_py_file:
         text_generator_py_file.write(text_generator_py_content)
 
-    create_repo(repo_id=title.replace(" ","-"), repo_type="space", space_sdk = "gradio")
+    create_repo(repo_id=tool_repo_id, repo_type="space", space_sdk = "gradio")
         #repo_type="space"
                
     # Upload the folder to the Hugging Face Hub
     upload_folder(
-        folder_path=title.replace(" ",""),
-        repo_id=f"Chris4K/{title.replace(" ","")}",
+        folder_path=tool_class,
+        repo_id=f"Chris4K/{tool_class}",
         repo_type="space"
     )
 
     # Return the generated files for download
-    return "README.md", "tool_config.json", "app.py", "requirements.txt", f"{title.replace(" ","_").lower()}.py"
+    return f"Chris4K/{tool_class}"
 
 
 # Define the inputs for the Gradio interface
 io = gr.Interface(generate_files,
                   inputs=[gr.inputs.Text(default="Text Generation Tool"), gr.inputs.Text(default="It takes an input named `prompt` which contains a system_role, user_message, context and history. It returns a text message.")],
-                  outputs=["text", "text", "text", "text", "text"])
+                  outputs=["text"])
 
 # Launch the Gradio interface
 io.launch()