Update app.py

app.py

@@ -1,21 +1,46 @@
 import os
 import gradio as gr
 import pandas as pd
+import requests
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+HF_MODEL = "bigscience/bloom-560m"
+HF_API_URL = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
+HF_API_TOKEN = os.getenv("HF_API_TOKEN")
 
 # --- Dummy Agent ---
 class DummyAgent:
     def __init__(self):
-        print("DummyAgent initialized (no SSL support).")
+        print("DummyAgent initialized.")
 
     def __call__(self, question: str) -> str:
         return "Error: Could not generate answer."
 
+# --- Hugging Face Inference API Agent ---
+class HFInferenceAgent:
+    def __init__(self):
+        if not HF_API_TOKEN:
+            raise ValueError("HF_API_TOKEN environment variable is not set.")
+        self.headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
+
+    def __call__(self, question: str) -> str:
+        payload = {
+            "inputs": question,
+            "parameters": {"max_new_tokens": 100, "return_full_text": False},
+            "options": {"use_cache": False}
+        }
+        try:
+            response = requests.post(HF_API_URL, headers=self.headers, json=payload, timeout=30)
+            response.raise_for_status()
+            result = response.json()
+            return result[0]['generated_text'].strip()
+        except Exception as e:
+            print(f"[HF Inference API Error] {e}")
+            return "Error: Could not generate answer."
+
 # --- Main Run & Submit Function ---
 def run_and_submit_all(profile: gr.OAuthProfile | None):
-    # Use your actual Space ID here as fallback if env not set
     space_id = os.getenv("SPACE_ID", "XeroDN/final_assignment")
 
     if profile:
@@ -27,41 +52,11 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     questions_url = f"{DEFAULT_API_URL}/questions"
     submit_url = f"{DEFAULT_API_URL}/submit"
 
+    # Initialize agent
     try:
-        import ssl
-        import requests
-        ssl_supported = True
-    except ImportError:
-        ssl_supported = False
-
-    if ssl_supported:
-        try:
-            from openai import OpenAI
-
-            class OpenAIAgent:
-                def __init__(self):
-                    self.client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-
-                def __call__(self, question: str) -> str:
-                    try:
-                        response = self.client.chat.completions.create(
-                            model="gpt-3.5-turbo",  # or "gpt-4" if you have access
-                            messages=[
-                                {"role": "system", "content": "You are a helpful and precise reasoning agent. Think step-by-step, and be concise."},
-                                {"role": "user", "content": question}
-                            ]
-                        )
-                        return response.choices[0].message.content.strip()
-                    except Exception as e:
-                        print(f"[OpenAI API Error] {e}")
-                        return "Error: Could not generate answer."
-
-            agent = OpenAIAgent()
-
-        except Exception as e:
-            print("Falling back to DummyAgent:", e)
-            agent = DummyAgent()
-    else:
+        agent = HFInferenceAgent()
+    except Exception as e:
+        print("Falling back to DummyAgent:", e)
         agent = DummyAgent()
 
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
@@ -118,7 +113,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
 # --- Gradio UI ---
 with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 GAIA Benchmark Agent Submission")
+    gr.Markdown("## 🤖 GAIA Benchmark Agent Submission (Hugging Face Inference API)")
     gr.Markdown("Click below to log in and run your agent on the GAIA benchmark.")
 
     gr.LoginButton()