Update app.py

app.py

@@ -3,7 +3,8 @@ import gradio as gr
 import requests
 import pandas as pd
 from dotenv import load_dotenv
-from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_openai import ChatOpenAI
+from langchain_nvidia_ai_endpoints import ChatNVIDIA
 
 # Load environment variables
 load_dotenv()
@@ -14,13 +15,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 class BasicAgent:
     def __init__(self):
-        self.llm = ChatGoogleGenerativeAI(
-            model="gemini-2.0-flash",
-            temperature=0.1,
-            max_tokens=1024,
-            api_key=os.getenv("GOOGLE_API_KEY"),
-            streaming=False  # Set to True if you want streaming output
-        )
+        self.llm = ChatOpenAI(model="gpt-4o-mini",streaming=False,base_url="https://aiproxy.sanand.workers.dev/openai/v1",api_key=os.getenv("AIPROXY_TOKEN"))
         self.instructions = """You are a helpful assistant. For every question, reply with only the answer—no explanation, "
             "no units, and no extra words. If the answer is a number, just return the number. "
             "If it is a word or phrase, return only that. If it is a list, return a comma-separated list with no extra words. "