update agent logic

agent.py

@@ -1,9 +1,25 @@
+import os
+import openai
+
 class BaseAgent:
     """Base class for all agents. Override __call__ to implement agent logic."""
     def __call__(self, question: str) -> str:
         raise NotImplementedError("Agent must implement __call__ method.")
 
-class ExampleAgent(BaseAgent):
-    """A simple agent that returns a fixed answer for demonstration purposes."""
+class LLMOpenAIAgent(BaseAgent):
+    """Agent that uses OpenAI's GPT-3.5-turbo to answer questions."""
+    def __init__(self, api_key=None):
+        self.api_key = api_key or os.getenv("OPENAI_API_KEY")
+        openai.api_key = self.api_key
+
     def __call__(self, question: str) -> str:
-        return "42" 
+        try:
+            response = openai.ChatCompletion.create(
+                model="gpt-3.5-turbo",
+                messages=[{"role": "user", "content": question}],
+                max_tokens=256,
+                temperature=0.2,
+            )
+            return response.choices[0].message["content"].strip()
+        except Exception as e:
+            return f"Error: {e}" 

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 import pandas as pd
-from agent import ExampleAgent
+from agent import LLMOpenAIAgent
 from api import APIClient
 
 # (Keep Constants as is)
@@ -20,7 +20,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         return "Please Login to Hugging Face with the button.", None
 
     api = APIClient(DEFAULT_API_URL)
-    agent = ExampleAgent()
+    agent = LLMOpenAIAgent()
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
 
     # Fetch questions

requirements.txt

@@ -1,2 +1,4 @@
 gradio
-requests
+requests
+openai
+transformers