Update app.py

app.py

@@ -1,27 +1,60 @@
 from flask import Flask, request, jsonify
-from openai import OpenAI
+import openai
 import os
-from flask_cors import CORS
-
 
 app = Flask(__name__)
-openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-CORS(app)
-assistant_id = "asst_R7FO44rNY25nJeSszxHktjAR"  # Replace with your actual Assistant ID
 
-@app.route('/chat', methods=['POST'])
+# Set your OpenAI API key
+openai.api_key = "sk-..."  # Replace with your actual key
+
+# Your Assistant ID (from platform.openai.com)
+ASSISTANT_ID = "asst_izF6HxamgRWJf8RnMf7tdoMA"
+
+# Store thread IDs per session/user if needed
+thread_store = {}
+
+@app.route("/chat", methods=["POST"])
 def chat():
-    user_message = request.json.get("message")
-    
-    thread = openai_client.beta.threads.create()
-    openai_client.beta.threads.messages.create(thread.id, role="user", content=user_message)
-    
-    run = openai_client.beta.threads.runs.create(thread_id=thread.id, assistant_id=assistant_id)
+    user_input = request.json.get("message")
+    user_id = request.json.get("user_id", "default")
+
+    # Create a thread if not exists
+    if user_id not in thread_store:
+        thread = openai.beta.threads.create()
+        thread_store[user_id] = thread.id
+
+    thread_id = thread_store[user_id]
+
+    # Step 1: Add message to thread
+    openai.beta.threads.messages.create(
+        thread_id=thread_id,
+        role="user",
+        content=user_input
+    )
+
+    # Step 2: Run the assistant
+    run = openai.beta.threads.runs.create(
+        thread_id=thread_id,
+        assistant_id=ASSISTANT_ID
+    )
+
+    # Step 3: Wait for completion (polling)
+    import time
+    while True:
+        run_status = openai.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
+        if run_status.status == "completed":
+            break
+        elif run_status.status == "failed":
+            return jsonify({"error": "Run failed"}), 500
+        time.sleep(1)
+
+    # Step 4: Fetch response
+    messages = openai.beta.threads.messages.list(thread_id=thread_id)
+    response_text = messages.data[0].content[0].text.value
+
+    return jsonify({"response": response_text})
 
-    while run.status != "completed":
-        run = openai_client.beta.threads.runs.retrieve(thread_id=thread.id, run_id=run.id)
 
-    messages = openai_client.beta.threads.messages.list(thread_id=thread.id)
-    bot_reply = messages.data[0].content[0].text.value
+if __name__ == "__main__":
+    app.run(debug=True)
 
-    return jsonify({"reply": bot_reply})