Update app.py

app.py

@@ -2,46 +2,37 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-import openai  # NEW: import OpenAI SDK
+import openai  # Added openai import
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Basic Agent Definition ---
-# Modified to use OpenAI GPT-4 API for Q&A
+# --- Basic Agent using OpenAI GPT ---
 class BasicAgent:
     def __init__(self):
-        self.api_key = os.getenv("OPEN_API_KEY")
-        if not self.api_key:
+        openai.api_key = os.getenv("OPENAI_API_KEY")
+        if not openai.api_key:
             raise ValueError("OPENAI_API_KEY environment variable is not set!")
-        openai.api_key = self.api_key
-        print("BasicAgent initialized with OpenAI API key.")
-    
+
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
         try:
             response = openai.ChatCompletion.create(
-                model="gpt-4",
+                model="gpt-4",  # or "gpt-3.5-turbo"
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant."},
                     {"role": "user", "content": question}
                 ],
-                max_tokens=200,
-                temperature=0
+                max_tokens=150,
+                temperature=0.7
             )
             answer = response['choices'][0]['message']['content'].strip()
-            print(f"Agent returning answer: {answer}")
             return answer
         except Exception as e:
-            print(f"Error calling OpenAI API: {e}")
+            print(f"OpenAI API error: {e}")
             return "Sorry, I couldn't answer that."
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
-    """
-    Fetches all questions, runs the BasicAgent on them, submits all answers,
-    and displays the results.
-    """
-    space_id = os.getenv("SPACE_ID")  # For link to your codebase
+    space_id = os.getenv("SPACE_ID")
 
     if profile:
         username = f"{profile.username}"
@@ -63,7 +54,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id else "Code repo link unavailable"
     print(agent_code)
 
-    # Fetch questions
     try:
         response = requests.get(questions_url, timeout=15)
         response.raise_for_status()
@@ -152,8 +142,8 @@ with gr.Blocks() as demo:
         3.  Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
         ---
         **Disclaimers:**
-        Once clicking on the "submit button, it can take quite some time ( this is the time for the agent to go through all the questions).
-        This space provides a basic setup and is intentionally sub-optimal to encourage you to develop your own, more robust solution. For instance for the delay process of the submit button, a solution could be to cache the answers and submit in a seperate action or even to answer the questions in async.
+        Once clicking on the "submit" button, it can take quite some time (this is the time for the agent to go through all the questions).
+        This space provides a basic setup and is intentionally sub-optimal to encourage you to develop your own, more robust solution. For instance, for the delay process of the submit button, a solution could be to cache the answers and submit in a separate action or even to answer the questions asynchronously.
         """
     )