Update app.py

app.py

@@ -1,38 +1,32 @@
 from flask import Flask, request, jsonify
-import requests
-import os
+from transformers import pipeline
+from flask_cors import CORS  # Import CORS
 
 app = Flask(__name__)
+CORS(app)  # Enable CORS for all routes
 
-# URL of the external chatbot API
-EXTERNAL_CHATBOT_URL = "https://ahil-mohammad.github.io/Bee-Server/"  # Replace with the correct URL
+# Alternatively, you can enable CORS for specific origins only (e.g., your GitHub Pages URL)
+# CORS(app, origins=["https://ahil-mohammad.github.io"])
 
-@app.route("/send_message", methods=["POST"])
+# Load your Hugging Face model (e.g., a conversational model)
+model = pipeline("conversational", model="microsoft/DialoGPT-medium")
+
+@app.route('/send_message', methods=['POST'])
 def send_message():
     try:
-        # Get the user message from the frontend
-        user_message = request.json.get("message", "")
+        # Get the incoming message from the request
+        data = request.get_json()
+        user_message = data['message']
         
-        # Send the message to the external chatbot API
-        external_response = requests.post(
-            EXTERNAL_CHATBOT_URL,
-            json={"message": user_message},
-            headers={"Content-Type": "application/json"}
-        )
-
-        # Check if the external API responded successfully
-        if external_response.status_code == 200:
-            # Get the chatbot's response
-            chatbot_response = external_response.json().get("response", "No response.")
-        else:
-            chatbot_response = f"Error from external chatbot: {external_response.status_code}"
-
-        # Send the chatbot response back to the frontend
-        return jsonify({"response": chatbot_response})
-
+        # Generate the model's response
+        response = model(user_message)
+        bot_reply = response[0]['generated_text']  # Assuming the model returns a list of conversations
+        
+        # Return the response as a JSON
+        return jsonify({'response': bot_reply})
+    
     except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        return jsonify({'error': str(e)}), 500
 
 if __name__ == "__main__":
-    # Default host and port for Hugging Face Spaces
-    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+    app.run(host="0.0.0.0", port=5000)