Update app.py

app.py

@@ -1,11 +1,22 @@
 import os
-import openai
+from openai import OpenAI
 from flask import Flask, request, jsonify, render_template
 
 app = Flask(__name__)
 
 # ✅ Get the API key from environment variables
-openai.api_key = os.getenv("OPENAI_API_KEY")
+api_key = os.getenv("OPENAI_API_KEY")
+
+if not api_key:
+    print("WARNING: OPENAI_API_KEY is not set. Chat functionality will not work.")
+    client = None
+else:
+    try:
+        from openai import OpenAI
+        client = OpenAI(api_key=api_key)
+    except Exception as e:
+        print(f"Error initializing OpenAI client: {e}")
+        client = None
 
 @app.route("/")
 def home():
@@ -15,7 +26,7 @@ def home():
 @app.route("/chat", methods=["POST"])
 def chat():
     try:
-        if not openai.api_key:
+        if not client:
             return jsonify({"error": "OpenAI API key not configured"}), 500
             
         data = request.json
@@ -24,27 +35,15 @@ def chat():
         if not user_message:
             return jsonify({"error": "Message is required"}), 400
 
-        # Try the new OpenAI client first, fall back to old method
-        try:
-            from openai import OpenAI
-            client = OpenAI()
-            response = client.chat.completions.create(
-                model="gpt-3.5-turbo",
-                messages=[{"role": "user", "content": user_message}],
-                temperature=0.7,
-                max_tokens=1000
-            )
-            reply = response.choices[0].message.content
-        except:
-            # Fallback to older OpenAI API method
-            response = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo",
-                messages=[{"role": "user", "content": user_message}],
-                temperature=0.7,
-                max_tokens=1000
-            )
-            reply = response.choices[0].message["content"]
+        # Use the new OpenAI client
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": user_message}],
+            temperature=0.7,
+            max_tokens=1000
+        )
 
+        reply = response.choices[0].message.content
         return jsonify({"reply": reply})
 
     except Exception as e:
@@ -53,7 +52,7 @@ def chat():
 @app.route("/health")
 def health():
     """Health check endpoint"""
-    return jsonify({"status": "healthy", "service": "SmartMate"})
+    return jsonify({"status": "healthy", "service": "ThutoAI"})
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)