Update app.py

app.py

@@ -1,30 +1,27 @@
 import gradio as gr
-import requests
+import openai
 import os
+from dotenv import load_dotenv
 
-# Retrieve the API key from the environment variables
+# Load environment variables from .env file
+load_dotenv()
+
+# Get the API key from the environment variable
 API_KEY = os.getenv("OPENAI_API_KEY")
-API_URL = "https://api.openai.com/v1/chat/completions"
+openai.api_key = API_KEY
 
-# Function to call the OpenAI API
+# Function to interact with OpenAI API
 def chat_with_model(prompt):
     if not API_KEY:
         return "Error: API key not found. Please set it in the environment variables."
     
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {API_KEY}"
-    }
-    payload = {
-        "model": "gpt-4",  # Replace with your desired model (e.g., gpt-3.5-turbo, gpt-4)
-        "messages": [{"role": "user", "content": prompt}]
-    }
-
     try:
-        response = requests.post(API_URL, json=payload, headers=headers)
-        response.raise_for_status()
-        return response.json()["choices"][0]["message"]["content"]
-    except requests.exceptions.RequestException as e:
+        response = openai.ChatCompletion.create(
+            model="gpt-4",  # Replace with your desired model
+            messages=[{"role": "user", "content": prompt}]
+        )
+        return response["choices"][0]["message"]["content"]
+    except openai.error.OpenAIError as e:
         return f"Error: {str(e)}"
 
 # Gradio Interface