Update app.py

app.py

@@ -1,50 +1,133 @@
-# Add this class replacing the current GeminiAgent
+import os
+import gradio as gr
+import requests
+import pandas as pd
+
+# Try to import Google GenAI SDK
+try:
+    from google import generativeai as genai
+except ImportError:
+    raise ImportError("Please install the 'google-generativeai' package: pip install google-generativeai")
+
+# --- Constants ---
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+
+# --- Multi-Key Gemini Agent ---
 class MultiAgent:
     def __init__(self):
         print("MultiAgent initialized with Gemini fallback.")
 
-        # Load primary Gemini key
-        self.gemini_key = os.getenv("GEMINI_API_KEY")
-        self.alt_key = os.getenv("GEMINI_API_KEY_2")  # Second key (optional)
+        self.primary_key = os.getenv("GEMINI_API_KEY")
+        self.secondary_key = os.getenv("GEMINI_API_KEY_2")
 
-        if not self.gemini_key:
+        if not self.primary_key:
             raise ValueError("GEMINI_API_KEY environment variable not set.")
 
-        import google.generativeai as genai
-        self.genai = genai
-
-        # Primary client
-        self.client_primary = genai.Client(api_key=self.gemini_key)
-        # Secondary client (fallback)
-        self.client_secondary = genai.Client(api_key=self.alt_key) if self.alt_key else None
-
+        genai.configure(api_key=self.primary_key)
         self.model_name = "gemini-2.5-flash-preview-04-17"
 
-    def _query_model(self, client, question):
+        self.client_primary = genai.GenerativeModel(self.model_name)
+        self.client_secondary = None
+        if self.secondary_key:
+            self.client_secondary = genai.GenerativeModel(self.model_name)
+
+    def query(self, client, question):
         prompt = f"Answer concisely without explanation: {question}"
-        response = client.models.generate_content(
-            model=self.model_name,
-            contents=prompt
-        )
+        response = client.generate_content(prompt)
         return response.text.strip().rstrip(".!?")
 
-    def __call__(self, question: str) -> str:
-        print(f"MultiAgent received question: {question[:50]}...")
+    def __call__(self, question):
+        print(f"🧠 GeminiAgent received question: {question[:60]}")
         try:
-            return self._query_model(self.client_primary, question)
+            return self.query(self.client_primary, question)
         except Exception as e:
             print(f"⚠️ Primary Gemini failed: {e}")
             if self.client_secondary:
                 try:
-                    print("Attempting fallback Gemini key...")
-                    return self._query_model(self.client_secondary, question)
+                    print("🔁 Trying fallback Gemini key...")
+                    genai.configure(api_key=self.secondary_key)
+                    self.client_secondary = genai.GenerativeModel(self.model_name)
+                    return self.query(self.client_secondary, question)
                 except Exception as e2:
                     print(f"❌ Secondary Gemini failed: {e2}")
-                    return "AGENT ERROR"
-            else:
-                return "AGENT ERROR"
-
-# Then replace:
-# agent = GeminiAgent()
-# with:
-# agent = MultiAgent()
+            return "AGENT ERROR"
+
+# --- Run and Submit ---
+def run_and_submit_all(profile: gr.OAuthProfile | None):
+    space_id = os.getenv("SPACE_ID")
+    if not profile:
+        return "Please Login to Hugging Face with the button.", None
+    username = f"{profile.username}"
+
+    questions_url = f"{DEFAULT_API_URL}/questions"
+    submit_url = f"{DEFAULT_API_URL}/submit"
+    try:
+        agent = MultiAgent()
+    except Exception as e:
+        return f"Agent initialization failed: {e}", None
+
+    agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
+    try:
+        response = requests.get(questions_url, timeout=15)
+        response.raise_for_status()
+        questions_data = response.json()
+        if not questions_data:
+            return "No questions received.", None
+    except Exception as e:
+        return f"Failed to fetch questions: {e}", None
+
+    results_log = []
+    answers_payload = []
+    for item in questions_data:
+        task_id = item.get("task_id")
+        question = item.get("question")
+        if not task_id or not question:
+            continue
+        try:
+            answer = agent(question)
+            answers_payload.append({"task_id": task_id, "submitted_answer": answer})
+            results_log.append({"Task ID": task_id, "Question": question, "Submitted Answer": answer})
+        except Exception as e:
+            results_log.append({"Task ID": task_id, "Question": question, "Submitted Answer": f"AGENT ERROR: {e}"})
+
+    if not answers_payload:
+        return "No answers submitted.", pd.DataFrame(results_log)
+
+    submission_data = {
+        "username": username.strip(),
+        "agent_code": agent_code,
+        "answers": answers_payload
+    }
+
+    try:
+        response = requests.post(submit_url, json=submission_data, timeout=60)
+        response.raise_for_status()
+        result = response.json()
+        status = (
+            f"✅ Submission Successful!\n"
+            f"User: {result.get('username')}\n"
+            f"Overall Score: {result.get('score', 'N/A')}% "
+            f"({result.get('correct_count', '?')}/{result.get('total_attempted', '?')} correct)\n"
+            f"Message: {result.get('message', '')}"
+        )
+        return status, pd.DataFrame(results_log)
+    except Exception as e:
+        return f"Submission failed: {e}", pd.DataFrame(results_log)
+
+# --- Gradio UI ---
+with gr.Blocks() as demo:
+    gr.Markdown("""# 🌟 Gemini Agent Evaluation
+    1. Login to Hugging Face
+    2. Click below to run your Gemini agent
+    3. Submit answers and view your score
+    """)
+    gr.LoginButton()
+    run_btn = gr.Button("Run Evaluation & Submit All Answers")
+    status_box = gr.Textbox(label="Submission Status", lines=6, interactive=False)
+    results_df = gr.DataFrame(label="Results Table", wrap=True)
+
+    run_btn.click(fn=run_and_submit_all, outputs=[status_box, results_df])
+
+if __name__ == "__main__":
+    print("\n--- App Starting ---")
+    demo.launch(debug=True, share=False)