Update app.py

app.py

@@ -2,26 +2,39 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from transformers import pipeline
+import openai  # NEW: import OpenAI SDK
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
-# Modified to use HF transformers pipeline for Q&A
+# Modified to use OpenAI GPT-4 API for Q&A
 class BasicAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
-        # Load the model pipeline for text2text-generation (question answering)
-        self.qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-small")
+        self.api_key = os.getenv("OPENAI_API_KEY")
+        if not self.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]}...")
-        # Run the question through the pipeline to get answer
-        output = self.qa_pipeline(question, max_length=64)
-        answer = output[0]['generated_text']
-        print(f"Agent returning answer: {answer}")
-        return answer
+        try:
+            response = openai.ChatCompletion.create(
+                model="gpt-4",
+                messages=[
+                    {"role": "system", "content": "You are a helpful assistant."},
+                    {"role": "user", "content": question}
+                ],
+                max_tokens=200,
+                temperature=0
+            )
+            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}")
+            return "Sorry, I couldn't answer that."
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """