Update app.py

app.py

@@ -10,42 +10,24 @@ client = InferenceClient("Futuresony/future_ai_12_10_2024.gguf")
 
 @app.route("/")
 def home():
-    return HTML_TEMPLATE
-
-function fetchMessageFromAI(message) {
-    document.getElementById('typing-indicator').style.display = 'flex';
-
-    fetch('/message', {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/json',
-        },
-        body: JSON.stringify({ text: message })
-    })
-    .then(response => {
-        if (!response.ok) {
-            throw new Error(`Server error: ${response.statusText}`);
-        }
-        return response.json();
-    })
-    .then(data => {
-        document.getElementById('typing-indicator').style.display = 'none';
-
-        if (data.response) {
-            addAIResponse(data.response, message);
-        } else {
-            console.error("No response from the server.");
-        }
-    })
-    .catch(error => {
-        document.getElementById('typing-indicator').style.display = 'none';
-        console.error('Error:', error);
-        appendMessage("An error occurred. Please try again later.", 'ai');
-    });
-}
-
+    # Render the HTML template
+    return render_template("index.html")
+
+@app.route("/message", methods=["POST"])
+def fetch_message():
+    data = request.json
+    message = data.get("text", "")
+    if not message:
+        return jsonify({"error": "No input provided."}), 400
+
+    # Process the message using the Hugging Face model
+    try:
+        response = client.text_generation(message)
+        return jsonify({"response": response})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":
     # Use PORT environment variable or default to 7860
     port = int(os.getenv("PORT", 7860))
-    app.run(host="0.0.0.0", port=port)
+    app.run(host="0.0.0.0", port=port)