Update app.py

app.py

@@ -63,9 +63,11 @@ class HybridAgent65:
 # ===========================
 # Run & Submit
 # ===========================
-def run_and_submit_all(profile):
-    if profile is None:
+def run_and_submit_all(profile_state):
+    profile = profile_state.value if isinstance(profile_state, gr.State) else profile_state
+    if not profile:
         return "❌ Please login with your Hugging Face account.", None
+
     username = profile["username"] if isinstance(profile, dict) else profile.username
     space_id = os.getenv("SPACE_ID", "unknown")
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
@@ -111,7 +113,7 @@ def run_and_submit_all(profile):
         return f"❌ Submission failed: {e}", pd.DataFrame(results_log)
 
 # ===========================
-# Gradio Interface (compatible with latest HF login)
+# Gradio Interface (stable for HF login)
 # ===========================
 with gr.Blocks() as demo:
     gr.Markdown("# 🎯 Hybrid GAIA Agent (65% Version)")
@@ -124,13 +126,18 @@ with gr.Blocks() as demo:
         """
     )
 
+    # State to store HF profile
+    user_state = gr.State()
+
+    # LoginButton stores profile into state
     login_btn = gr.LoginButton()
+    login_btn.click(lambda p: p, inputs=[login_btn], outputs=[user_state])
+
     run_button = gr.Button("🚀 Run Evaluation & Submit All Answers")
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
     
-    # HF login returns profile dict,直接作為 run 函數輸入
-    run_button.click(fn=run_and_submit_all, inputs=[login_btn], outputs=[status_output, results_table])
+    run_button.click(fn=run_and_submit_all, inputs=[user_state], outputs=[status_output, results_table])
 
 if __name__ == "__main__":
     demo.launch(debug=True, share=False)