Create OpenAI class as last changes on this file had not been saved

app.py

@@ -1,4 +1,5 @@
 import os
+import openai
 import gradio as gr
 import requests
 import inspect
@@ -9,11 +10,21 @@ from smolagents import CodeAgent, HfApiModel
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-HF_TOKEN = os.getenv("HF_TOKEN")
-model = HfApiModel(api_key=HF_TOKEN)
+OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
 
 
-tools = []
+class OpenAIModel:
+    def __init__(self):
+        openai.api_key = OPEN_AI_KEY
+    
+    def run(self, prompt: str) -> str:
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # or "gpt-4"
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=300
+        )
+        return response["choices"][0]["message"]["content"]
+
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -51,6 +62,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
+        model = OpenAIModel()
         agent = BasicAgent(tools=tools, model=model, planning_interval=3)
     except Exception as e:
         print(f"Error instantiating agent: {e}")