Update app.py

app.py

@@ -1,12 +1,14 @@
 from flask import Flask, request, jsonify
-import openai
+import requests
 
 # Initialize Flask application
 app = Flask(__name__)
 
-# Set the base URL for the API endpoint and API key
-openai.api_base = "https://api.pawan.krd/unfiltered/v1"
-openai.api_key = "pk-IDpTrFbbikTWTSUdJsckGStERwgiZCdaCznbKagvOAVxAftU"
+# Set the URL for the Cosmo-RP API
+url = "https://api.pawan.krd/cosmosrp/v1/chat/completions"
+headers = {
+    "Content-Type": "application/json"
+}
 
 history = []
 first_message = True
@@ -42,17 +44,22 @@ def chat_bot():
         first_message = False
     history.append({"role": "user", "content": message})
 
-    # Use the Cosmo-RP model for the completion request
+    # Prepare the data for the Cosmo-RP API call
+    api_data = {
+        "model": "cosmosrp",
+        "messages": [{"role": "system", "content": system_message}] + [{"role": "user", "content": message}]
+    }
+
+    # Make the POST request to the Cosmo-RP API
     try:
-        response = openai.Completion.create(
-            model="Cosmo-RP",
-            prompt="\n".join([msg['content'] for msg in history]),
-            max_tokens=150
-        ).choices[0].text.strip()
+        response = requests.post(url, headers=headers, json=api_data)
+        response.raise_for_status()  # Raise an error for bad responses
+        api_response = response.json()
+        assistant_response = api_response.get("choices", [{}])[0].get("message", {}).get("content", "No response")
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-    history.append({"role": "assistant", "content": response})
+    history.append({"role": "assistant", "content": assistant_response})
 
     # Trim history to keep within the word limit
     trim_history_to_word_limit(history, word_limit)
@@ -60,9 +67,9 @@ def chat_bot():
     # Save to a text file
     with open(f"{file_name}.txt", 'a', encoding='utf-8') as f:
         f.write(f"User: {message}\n")
-        f.write(f"{chatbot_name}: {response}\n")
+        f.write(f"{chatbot_name}: {assistant_response}\n")
     
-    return jsonify({"response": response, "history": history})
+    return jsonify({"response": assistant_response, "history": history})
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=5000)