Update app.py

app.py

@@ -114,7 +114,6 @@ def configure_gemini(api_key: str | None, model_name: str | None) -> str:
         return "Please select a Gemini model."
     try:
         genai.configure(api_key=api_key)
-        # Test using the selected model
         genai.GenerativeModel(model_name).generate_content("ping")
         return f"Gemini configured successfully with **{model_name}**."
     except Exception as e:
@@ -132,7 +131,7 @@ def call_gemini(prompt: str, api_key: str, model_name: str) -> str:
         return f"Error calling Gemini API with {model_name}: {e}"
 
 
-# --- AI workflow logic (uses selected model) ---
+# --- AI workflow logic ---
 
 def ai_workflow_chat(
     message: str,
@@ -155,7 +154,6 @@ def ai_workflow_chat(
     str,
     str
 ]:
-    # Append user message
     history.append([message, None])
     bot_message = ""
     new_repo_id = repo_id_state
@@ -164,13 +162,98 @@ def ai_workflow_chat(
     updated_container = container_logs
     updated_build = build_logs
 
-    # -- same workflow logic as before, but use call_gemini(prompt, gemini_api_key, gemini_model) --
-
-    # example when generating code:
-    # resp = call_gemini(prompt, gemini_api_key, gemini_model)
+    try:
+        if not hf_profile or not hf_token:
+            bot_message = "Please log in to Hugging Face first."
+            new_workflow = "awaiting_login"
+        elif not gemini_api_key or not gemini_model:
+            bot_message = "Please enter your API key and select a Gemini model."
+            new_workflow = "awaiting_api_key"
+        elif (new_workflow == "idle" or "create" in message.lower()) and not new_repo_id:
+            bot_message = "What should the Space be called? (e.g., `my-awesome-app`)"
+            new_workflow = "awaiting_repo_name"
+        elif new_workflow == "awaiting_repo_name":
+            repo_name = message.strip()
+            if not repo_name:
+                bot_message = "Please provide a valid Space name."
+            else:
+                bot_message = f"Creating Space `{hf_profile.username}/{repo_name}`..."
+                new_repo_id, iframe_html = create_space_action(repo_name, space_sdk, hf_profile, hf_token)
+                updated_preview = iframe_html
+                bot_message += "\n✅ Space created."
+                new_workflow = "awaiting_app_description"
+        elif new_workflow in ("awaiting_app_description", "debugging"):
+            if new_workflow == "awaiting_app_description":
+                app_desc = message
+                bot_message = f"Generating code for a `{space_sdk}` app based on: '{app_desc}'..."
+                prompt = f"""
+You are an AI assistant specializing in Hugging Face Spaces using the {space_sdk} SDK.
+Generate a full, single-file Python app based on:
+'{app_desc}'
+Return **only** the code block (```python ...```).
+"""
+            else:
+                debug_instr = message
+                logs = get_container_logs_action(new_repo_id, hf_profile, hf_token)
+                bot_message = f"Analyzing logs and applying fixes: '{debug_instr}'..."
+                prompt = f"""
+You are debugging a {space_sdk} Space.
+Logs:
+{logs}
+User instructions:
+'{debug_instr}'
+Generate a fixed, single-file Python app. Return only the ```python``` code block.
+"""
+            new_workflow = "generating_code"
+            resp = call_gemini(prompt, gemini_api_key, gemini_model)
+            start = resp.find("```python")
+            end   = resp.rfind("```")
+            if start != -1 and end != -1 and end > start:
+                code = resp[start + len("```python"):end].strip()
+                bot_message += "\n✅ Code generated. Uploading..."
+                new_workflow = "uploading_code"
+                upload_log = upload_file_to_space_action(code, "app.py", new_repo_id, hf_profile, hf_token)
+                bot_message += "\n" + upload_log
+                if "✅ Uploaded" in upload_log:
+                    bot_message += "\nThe Space is now rebuilding. Say 'check logs' to fetch them."
+                    new_workflow = "awaiting_log_check"
+                    updated_preview = f'<iframe src="https://huggingface.co/spaces/{new_repo_id}" width="100%" height="500px"></iframe>'
+                else:
+                    new_workflow = "idle"
+            else:
+                bot_message += f"\n⚠️ Could not parse code from Gemini.\nResponse:\n{resp}"
+                new_workflow = "awaiting_app_description"
+        elif new_workflow == "awaiting_log_check" and "check logs" in message.lower():
+            bot_message = "Fetching logs..."
+            updated_container = get_container_logs_action(new_repo_id, hf_profile, hf_token)
+            updated_build     = get_build_logs_action(new_repo_id, hf_profile, hf_token)
+            bot_message += "\n✅ Logs updated. Describe any errors or say 'generate fix'."
+            new_workflow = "reviewing_logs"
+        elif new_workflow == "reviewing_logs" and "generate fix" in message.lower():
+            latest = get_container_logs_action(new_repo_id, hf_profile, hf_token)
+            if "Error" not in latest and "Exception" not in latest:
+                bot_message = "No clear error found. What should I fix?"
+                new_workflow = "reviewing_logs"
+            else:
+                bot_message = "Generating a fix based on logs..."
+                new_workflow = "debugging"
+        elif "reset" in message.lower():
+            bot_message = "Workflow reset."
+            new_repo_id = None
+            updated_preview = "<p>No Space created yet.</p>"
+            updated_container = ""
+            updated_build = ""
+            new_workflow = "idle"
+        else:
+            bot_message = "Command not recognized. Try 'create', 'check logs', 'generate fix', or 'reset'."
+    except Exception as e:
+        bot_message = f"Unexpected error: {e}"
+        new_workflow = "idle"
 
-    # [Omitted for brevity; insert your existing logic here, replacing calls to
-    # call_gemini(prompt, gemini_api_key) with call_gemini(prompt, gemini_api_key, gemini_model).]
+    if history and history[-1][1] is None:
+        history[-1][1] = bot_message
+    else:
+        history.append([None, bot_message])
 
     return history, new_repo_id, new_workflow, updated_preview, updated_container, updated_build
 
@@ -178,13 +261,13 @@ def ai_workflow_chat(
 # --- Build the Gradio UI ---
 
 with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
-    hf_profile      = gr.State(None)
-    hf_token        = gr.State(None)
-    gemini_key      = gr.State(None)
-    gemini_model    = gr.State("gemini-2.5-pro-preview-03-25")
-    repo_id         = gr.State(None)
-    workflow        = gr.State("idle")
-    sdk_state       = gr.State("gradio")
+    hf_profile   = gr.State(None)
+    hf_token     = gr.State(None)
+    gemini_key   = gr.State(None)
+    gemini_model = gr.State("gemini-2.5-pro-preview-03-25")
+    repo_id      = gr.State(None)
+    workflow     = gr.State("idle")
+    sdk_state    = gr.State("gradio")
 
     with gr.Row():
         # Sidebar
@@ -193,11 +276,9 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             login_status = gr.Markdown("*Not logged in.*")
             login_btn    = gr.LoginButton(variant="huggingface")
 
-            # init & update login status
             ai_builder_tab.load(show_profile, outputs=login_status)
             login_btn.click(show_profile, outputs=login_status)
-            login_btn.click(lambda profile, token: (profile, token),
-                            outputs=[hf_profile, hf_token])
+            login_btn.click(lambda p, t: (p, t), outputs=[hf_profile, hf_token])
 
             gr.Markdown("## Google AI Studio API Key")
             gemini_input  = gr.Textbox(label="API Key", type="password")
@@ -215,8 +296,8 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             )
             model_selector.change(lambda m: m, inputs=model_selector, outputs=gemini_model)
 
-            # configure Gemini whenever key or model changes
-            gr.Row().load(
+            # configure Gemini on key or model change
+            ai_builder_tab.load(
                 configure_gemini,
                 inputs=[gemini_key, gemini_model],
                 outputs=[gemini_status]
@@ -242,7 +323,6 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             user_input = gr.Textbox(placeholder="Type your message…")
             send_btn   = gr.Button("Send", interactive=False)
 
-            # enable send only when logged in & key & model selected
             ai_builder_tab.load(
                 lambda p, k, m: gr.update(interactive=bool(p and k and m)),
                 inputs=[hf_profile, gemini_key, gemini_model],
@@ -269,11 +349,10 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             run_txt   = gr.Textbox(label="Container Logs", lines=10, interactive=False)
 
             def wrap_chat(msg, history, prof, tok, key, model, rid, wf, sdk, prev, run_l, build_l):
-                out = ai_workflow_chat(
+                new_hist, new_rid, new_wf, new_prev, new_run, new_build = ai_workflow_chat(
                     msg, history, prof, tok, key, model, rid, wf, sdk, prev, run_l, build_l
                 )
-                hist, new_rid, new_wf, new_prev, new_run, new_build = out
-                return [(u or "", v or "") for u, v in hist], new_rid, new_wf, new_prev, new_run, new_build
+                return [(u or "", v or "") for u, v in new_hist], new_rid, new_wf, new_prev, new_run, new_build
 
             send_btn.click(
                 wrap_chat,
@@ -292,7 +371,93 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             )
 
 with gr.Blocks(title="Manual Hugging Face Space Manager") as manual_control_tab:
-    # ... (manual tab unchanged) ...
+    manual_profile = gr.State(None)
+    manual_token   = gr.State(None)
+    manual_repo    = gr.State(None)
+
+    gr.Markdown("## Manual Sign-In & Space Management")
+    manual_login_btn = gr.LoginButton(variant="huggingface", size="lg")
+    manual_status    = gr.Markdown("*Not logged in.*")
+    manual_models    = gr.Markdown()
+
+    manual_control_tab.load(show_profile, outputs=manual_status)
+    manual_login_btn.click(show_profile, outputs=manual_status)
+    manual_control_tab.load(list_private_models, outputs=manual_models)
+    manual_login_btn.click(list_private_models, outputs=manual_models)
+    manual_login_btn.click(lambda p, t: (p, t), outputs=[manual_profile, manual_token])
+
+    manual_repo_name = gr.Textbox(label="New Space name", placeholder="my-space")
+    manual_sdk_sel   = gr.Radio(choices=["gradio","streamlit"], value="gradio", label="Template SDK")
+    manual_create_btn  = gr.Button("Create Space", interactive=False)
+    manual_create_logs = gr.Textbox(label="Create Logs", lines=3, interactive=False)
+    manual_preview     = gr.HTML("<p>No Space created yet.</p>")
+
+    manual_control_tab.load(
+        lambda p, t: gr.update(interactive=bool(p and t)),
+        inputs=[manual_profile, manual_token],
+        outputs=[manual_create_btn]
+    )
+    manual_login_btn.click(
+        lambda p, t: gr.update(interactive=bool(p and t)),
+        inputs=[manual_profile, manual_token],
+        outputs=[manual_create_btn]
+    )
+
+    manual_create_btn.click(
+        create_space_action,
+        inputs=[manual_repo_name, manual_sdk_sel, manual_profile, manual_token],
+        outputs=[manual_repo, manual_preview]
+    ).then(lambda _: "", outputs=[manual_create_logs])
+
+    manual_path = gr.Textbox(label="Path in Space", value="app.py")
+    manual_file = gr.File(label="Select file")
+    manual_up_btn = gr.Button("Upload File", interactive=False)
+    manual_up_log = gr.Textbox(label="Upload Logs", lines=2, interactive=False)
+
+    manual_control_tab.load(
+        lambda rid, p, t: gr.update(interactive=bool(rid and p and t)),
+        inputs=[manual_repo, manual_profile, manual_token],
+        outputs=[manual_up_btn]
+    )
+    manual_login_btn.click(
+        lambda rid, p, t: gr.update(interactive=bool(rid and p and t)),
+        inputs=[manual_repo, manual_profile, manual_token],
+        outputs=[manual_up_btn]
+    )
+
+    manual_up_btn.click(
+        upload_file_to_space_action,
+        inputs=[manual_file, manual_path, manual_repo, manual_profile, manual_token],
+        outputs=[manual_up_log]
+    )
+
+    manual_build_btn     = gr.Button("Fetch Build Logs", interactive=False)
+    manual_container_btn = gr.Button("Fetch Container Logs", interactive=False)
+    manual_build_txt     = gr.Textbox(label="Build Logs", lines=10, interactive=False)
+    manual_container_txt = gr.Textbox(label="Container Logs", lines=10, interactive=False)
+
+    for btn in (manual_build_btn, manual_container_btn):
+        manual_control_tab.load(
+            lambda rid, p, t: gr.update(interactive=bool(rid and p and t)),
+            inputs=[manual_repo, manual_profile, manual_token],
+            outputs=[btn]
+        )
+        manual_login_btn.click(
+            lambda rid, p, t: gr.update(interactive=bool(rid and p and t)),
+            inputs=[manual_repo, manual_profile, manual_token],
+            outputs=[btn]
+        )
+
+    manual_build_btn.click(
+        get_build_logs_action,
+        inputs=[manual_repo, manual_profile, manual_token],
+        outputs=[manual_build_txt]
+    )
+    manual_container_btn.click(
+        get_container_logs_action,
+        inputs=[manual_repo, manual_profile, manual_token],
+        outputs=[manual_container_txt]
+    )
 
 demo = gr.TabbedInterface(
     [ai_builder_tab, manual_control_tab],