fixes on handle_upload

README.md

@@ -29,6 +29,7 @@ app_file: app.py
 **An AI Agentic Tool Server for GitHub Issue Triaging & Analysis**
 
 [![MCP's 1st Birthday Hackathon](https://img.shields.io/badge/MCP%27s%201st%20Birthday-Hackathon-blue)](https://huggingface.co/MCP-1st-Birthday)
+[![Powered by Gradio](https://img.shields.io/badge/Powered%20by-Gradio-orange)](https://gradio.app/)
 
 
 > **🎯 Track 1 Submission**: Building MCP (Enterprise)

app.py

@@ -1052,8 +1052,7 @@ def generate_theme(
         llm_api_key (str, optional): API Key. **OPTIONAL**. If empty, uses server keys.
         theme_mode (str, optional): Theme mode. Defaults to 'Designer'. Use 'Engineer' if you want to specify the details and 'Designer' if you want the model to handle that; in this mode, use short prompts.
     """
-    global generated_theme_state
-
+    
     _validate_api_keys(llm_api_key, request)
 
     try:
@@ -1160,8 +1159,7 @@ def generate_theme(
         set_call = f".set(\n    {set_args}\n)" if set_kwargs else ""
         py_code = f"import gradio as gr\n\ntheme = gr.themes.Default(\n    {init_args}\n){set_call}"
 
-        generated_theme_state = theme_object
-        return py_code, str(status_message), css_str
+        return py_code, theme_object, str(status_message), css_str
 
     except Exception as e:
         import traceback
@@ -1179,23 +1177,37 @@ def generate_theme(
         raise gr.Error(detailed_msg)
 
 
-def handle_upload(theme_name, hf_token):
+def handle_upload(theme_name, generated_theme, hf_token):    
     """
-    Handles uploading the generated theme.
+    Handles uploading the generated theme to Hugging Face Hub.
+    This function takes a generated theme and pushes it to the Hugging Face Hub repository.
+    It validates that a theme has been generated and that the required credentials are provided,
+    then attempts to upload the theme and returns a success or error message.
+    
+    Args:
+        theme_name (str): The name of the theme to be uploaded. This will be used as the repository name
+        on Hugging Face Hub. Must not be empty.
+        generated_theme (object): Theme generated object
+        hf_token (str): The Hugging Face API token for authentication. Required to push the theme to the hub.
+        Must not be empty.
+    
+    Returns    
+        None
     """
-    global generated_theme_state
-
-    if not isinstance(generated_theme_state, gr.themes.Default):
-        return gr.Markdown("⚠️ Please generate a theme first.", visible=True)
+    
+    if not isinstance(generated_theme, gr.themes.Default):
+        gr.Info("⚠️ Please generate a theme first.")
+        return
     if not theme_name or not hf_token:
-        return gr.Markdown("⚠️ Please fill in the theme name and token.", visible=True)
+        gr.Info("⚠️ Please fill in the theme name and token.")
+        return
     try:
-        url = generated_theme_state.push_to_hub(repo_name=theme_name, hf_token=hf_token, theme_name=theme_name)
+        url = generated_theme.push_to_hub(repo_name=theme_name, hf_token=hf_token, theme_name=theme_name)
         space_id = "/".join(url.split("/")[-2:])
-        success_message = f"✅ **Success!** Theme uploaded to: [{url}]({url})\n<br>To use: `theme='{space_id}'`"
-        return gr.Markdown(success_message, visible=True)
+        gr.Info(f"✅ Success! Theme uploaded to: ({url}), To use: `theme='{space_id}'`")
+        return
     except Exception as e:
-        return gr.Markdown(f"❌ **Error:** {e}", visible=True)
+        raise gr.Error(f"❌ **Error:** {e}", visible=True)
 
 
 def _process_sketch_logic(final_image, text_description, llm_provider, llm_model, llm_api_key):
@@ -1372,7 +1384,7 @@ def _create_example_app():
                     "number",
                     "textbox",
                     title="Interface Component (Compact)",
-                    # show_api=False,
+                    # api_name=False,
                     api_name=False,
                 )
                 with gr.Row():
@@ -1553,7 +1565,7 @@ css = """
 #app { height: 100vh; overflow-y: scroll; }
 .min-h-50px { min-height: 50px; }
 """
-generated_theme_state = None  # Setting global instead gr.State()
+
 theme = gr.themes.Default(
     primary_hue='blue',
     secondary_hue='teal',
@@ -1612,7 +1624,7 @@ theme = gr.themes.Default(
 
 with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MCP Server)") as app:
     css_data_transport = gr.Textbox(visible=False)
-
+    generated_theme_state = gr.State(value=None)
     gr.HTML(_get_mcp_header())
 
     # GLOBAL CONFIGURATION
@@ -1762,8 +1774,7 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
                         with gr.TabItem("Publish to Hub"):
                             hub_theme_name = gr.Textbox(label="Theme Name", placeholder="my-amazing-theme")
                             hub_hf_token = gr.Textbox(label="HF Token", type="password", placeholder="hf_...")
-                            upload_button = gr.Button("Upload 🚀")
-                            upload_status = gr.Markdown(visible=False)
+                            upload_button = gr.Button("Upload 🚀")                            
                 chatbot_comp, msg_input_comp, add_msg_btn_comp = _create_example_app()
 
         with gr.TabItem("BONUS 2 - Sketch Image to Gradio UI"):
@@ -1827,7 +1838,7 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
         fn=update_all_model_choices,
         inputs=[global_provider],
         outputs=[global_model, sketch_model_dropdown],
-        show_api=False,
+        api_name=False,
     )
 
     # Tool 1: Sync
@@ -1852,7 +1863,7 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
         outputs=[issue_report_output, issue_thought_output],
         api_name="analyze_github_issue",
         show_progress_on=[issue_report_output],
-    ).then(fn=change_cot_visibility, inputs=[issue_thought_output], outputs=[issue_cot_accordion], show_api=False)
+    ).then(fn=change_cot_visibility, inputs=[issue_thought_output], outputs=[issue_cot_accordion], api_name=False)
 
     # Tool 3: Duplicate Finder
     dup_find_button.click(
@@ -1872,7 +1883,7 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
         fn=change_cot_visibility,
         inputs=[dup_thought_output],
         outputs=[dup_accordion],
-        show_api=False,
+        api_name=False,
     )
 
     # Tool 4: Prioritizer
@@ -1882,7 +1893,7 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
         outputs=[prio_report_output, prio_thought_output],
         api_name="prioritize_open_issues",
         show_progress_on=[prio_report_output],
-    ).success(fn=change_cot_visibility, inputs=[prio_thought_output], outputs=[prio_cot_accordion], show_api=False)
+    ).success(fn=change_cot_visibility, inputs=[prio_thought_output], outputs=[prio_cot_accordion], api_name=False)
 
     # Tool 5: Action Runner
     action_button.click(
@@ -1933,7 +1944,7 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
             global_api_key,
             theme_mode,
         ],
-        outputs=[python_code_output, status_output, css_data_transport],
+        outputs=[python_code_output, generated_theme_state, status_output, css_data_transport],
     ).then(
         fn=None,
         inputs=[css_data_transport],
@@ -1942,8 +1953,8 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
     )
     upload_button.click(
         fn=handle_upload,
-        inputs=[hub_theme_name, hub_hf_token],
-        outputs=[upload_status],
+        inputs=[hub_theme_name, generated_theme_state, hub_hf_token],        
+        api_name=False
     )
 
     # Bonus 2: Sketch
@@ -1951,10 +1962,10 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
         fn=generate_sketch_ui,
         inputs=[sketch_input, text_desc_input, global_provider, sketch_model_dropdown, global_api_key],
         outputs=[sketch_code_output],
-        show_api=False,
+        api_name=False,
     )
 
-    ui_event.success(fn=_create_gradio_lite_html, inputs=[sketch_code_output], outputs=[sketch_preview_output], show_api=False)
+    ui_event.success(fn=_create_gradio_lite_html, inputs=[sketch_code_output], outputs=[sketch_preview_output], api_name=False)
 
     api_trigger_btn.click(
         fn=generate_sketch_to_ui,
@@ -1985,8 +1996,8 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
         fn=add_message_to_chat,
         inputs=[chatbot_comp, msg_input_comp],
         outputs=[chatbot_comp],
-        show_api=False,
-    ).then(fn=lambda: "", outputs=[msg_input_comp], show_api=False)
+        api_name=False,
+    ).then(fn=lambda: "", outputs=[msg_input_comp], api_name=False)
 
     # Init
     app.load(fn=_initialize_database, inputs=None, outputs=None, show_api=False)
@@ -1994,4 +2005,4 @@ with gr.Blocks(theme=theme, css=css, head=THEME_JS, title="GitRepo Inspector (MC
 # endregion
 if __name__ == "__main__":
     app.allowed_paths = ["."]
-    app.launch(mcp_server=True, share=True, show_error=True, server_port=7860, enable_monitoring=False)
+    app.launch(mcp_server=True, share=True, show_error=True, server_port=7860)