Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, send_from_directory
 import requests
 
 # 🔧 Hugging Face inference endpoint and token
@@ -20,18 +20,36 @@ SYSTEM_PROMPT = (
 app = Flask(__name__)
 chat_history = []
 
+@app.route("/")
+def index():
+    return send_from_directory(".", "index.html")
+
+@app.route("/script.js")
+def serve_script():
+    return send_from_directory(".", "script.js")
+
+@app.route("/style.css")
+def serve_css():
+    return send_from_directory(".", "style.css")
+
 @app.route("/chat", methods=["POST"])
 def chat():
     global chat_history
     user_input = request.json.get("message", "")
+    if not user_input:
+        return jsonify({"error": "Empty message"}), 400
+
     chat_history.append({"role": "user", "content": user_input})
 
     payload = {
         "inputs": [
             {"role": "system", "content": SYSTEM_PROMPT},
-            *chat_history[-10:]  # recent messages
+            *chat_history[-10:]
         ],
-        "parameters": {"max_new_tokens": 250, "temperature": 0.7}
+        "parameters": {
+            "max_new_tokens": 250,
+            "temperature": 0.7
+        }
     }
 
     headers = {
@@ -44,16 +62,14 @@ def chat():
     if response.status_code != 200:
         return jsonify({"error": "Model error", "details": response.text}), 500
 
-    result = response.json()
-    reply = result.get("generated_text", "") or result[0]["generated_text"]
-    chat_history.append({"role": "assistant", "content": reply})
+    try:
+        result = response.json()
+        reply = result.get("generated_text", "") or result[0]["generated_text"]
+    except Exception as e:
+        return jsonify({"error": "Unexpected model response", "details": str(e)}), 500
 
+    chat_history.append({"role": "assistant", "content": reply})
     return jsonify({"response": reply})
 
-@app.route("/")
-def index():
-    return "SHODAN backend is running."
-
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)
-