Update app.py

app.py

@@ -1,7 +1,6 @@
-# app.py (Updated for OpenAI v1.x)
+# app.py (Compatible with OpenAI v0.x and v1.x)
 import gradio as gr
 import openai
-from openai import OpenAI  # Import the new client
 import threading
 import time
 import numpy as np
@@ -10,20 +9,28 @@ import os
 import pickle
 from datetime import datetime
 
-# Initialize OpenAI client
-client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
-
 # === CONFIG ===
 EMBEDDING_MODEL = "text-embedding-3-small"
 CHAT_MODEL = "gpt-4o"  # Updated to current model
 MEMORY_FILE = "memory.pkl"
 INDEX_FILE = "memory.index"
 
+# Initialize OpenAI API
+openai.api_key = os.environ.get("OPENAI_API_KEY")
+
 # === EMBEDDING UTILS ===
 def get_embedding(text, model=EMBEDDING_MODEL):
     text = text.replace("\n", " ")
-    response = client.embeddings.create(input=[text], model=model)
-    return response.data[0].embedding
+    
+    # Compatible API call for both v0.x and v1.x
+    try:
+        # Try v1.x API first
+        response = openai.embeddings.create(input=[text], model=model)
+        return response.data[0].embedding
+    except AttributeError:
+        # Fallback to v0.x API
+        response = openai.Embedding.create(input=[text], model=model)
+        return response['data'][0]['embedding']
 
 def cosine_similarity(vec1, vec2):
     vec1 = np.array(vec1)
@@ -51,20 +58,32 @@ conversation = []
 turn_count = 0
 auto_mode = False
 
-# === CHAT COMPLETION (UPDATED FOR OPENAI v1.x) ===
+# === CHAT COMPLETION (Compatible with both APIs) ===
 def chat_completion(system, messages, model=CHAT_MODEL):
     try:
         # Build message list with system prompt
         full_messages = [{"role": "system", "content": system}] 
         full_messages.extend(messages)
         
-        response = client.chat.completions.create(
-            model=model,
-            messages=full_messages,
-            temperature=0.7,
-            max_tokens=150
-        )
-        return response.choices[0].message.content.strip()
+        # Try v1.x API first
+        try:
+            response = openai.chat.completions.create(
+                model=model,
+                messages=full_messages,
+                temperature=0.7,
+                max_tokens=150
+            )
+            return response.choices[0].message.content.strip()
+        except AttributeError:
+            # Fallback to v0.x API
+            response = openai.ChatCompletion.create(
+                model=model,
+                messages=full_messages,
+                temperature=0.7,
+                max_tokens=150
+            )
+            return response['choices'][0]['message']['content'].strip()
+            
     except Exception as e:
         return f"[API Error: {str(e)}]"