Update app.py

app.py

@@ -1,32 +1,38 @@
 from flask import Flask, request, jsonify
-from transformers import AutoModelForCausalLM, AutoTokenizer
+import requests
 import os
 
 app = Flask(__name__)
 
-# Load model and tokenizer
-MODEL_NAME = "bigscience/bloomz-1b1"  # Replace with your model name
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
+# URL of the external chatbot API
+EXTERNAL_CHATBOT_URL = "https://your-external-chatbot-url.com/api/chat"  # Replace with the correct URL
 
-@app.route("/api/chat", methods=["POST"])
-def chat():
+@app.route("/send_message", methods=["POST"])
+def send_message():
     try:
-        # Get user message from the POST request
+        # Get the user message from the frontend
         user_message = request.json.get("message", "")
         
-        # Tokenize user input
-        inputs = tokenizer(user_message, return_tensors="pt", truncation=True, max_length=512)
-        
-        # Generate response
-        outputs = model.generate(inputs["input_ids"], max_length=150, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
-        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        
-        # Return the response as JSON
-        return jsonify({"response": response})
+        # 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})
+
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":
-    # Use the default port for Hugging Face Spaces
+    # Default host and port for Hugging Face Spaces
     app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))