Update app.py

app.py

@@ -1,6 +1,7 @@
-# app.py (Updated for 3-Chatbot System)
+# app.py (Updated for OpenAI v1.x)
 import gradio as gr
 import openai
+from openai import OpenAI  # Import the new client
 import threading
 import time
 import numpy as np
@@ -9,8 +10,8 @@ import os
 import pickle
 from datetime import datetime
 
-# Set API key
-openai.api_key = os.environ.get("OPENAI_API_KEY")
+# Initialize OpenAI client
+client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
 # === CONFIG ===
 EMBEDDING_MODEL = "text-embedding-3-small"
@@ -21,7 +22,7 @@ INDEX_FILE = "memory.index"
 # === EMBEDDING UTILS ===
 def get_embedding(text, model=EMBEDDING_MODEL):
     text = text.replace("\n", " ")
-    response = openai.embeddings.create(input=[text], model=model)
+    response = client.embeddings.create(input=[text], model=model)
     return response.data[0].embedding
 
 def cosine_similarity(vec1, vec2):
@@ -50,12 +51,16 @@ conversation = []
 turn_count = 0
 auto_mode = False
 
-# === CHAT COMPLETION ===
+# === CHAT COMPLETION (UPDATED FOR OPENAI v1.x) ===
 def chat_completion(system, messages, model=CHAT_MODEL):
     try:
-        response = openai.chat.completions.create(
+        # Build message list with system prompt
+        full_messages = [{"role": "system", "content": system}] 
+        full_messages.extend(messages)
+        
+        response = client.chat.completions.create(
             model=model,
-            messages=[{"role": "system", "content": system}] + messages,
+            messages=full_messages,
             temperature=0.7,
             max_tokens=150
         )