Update app.py

app.py

@@ -1,42 +1,108 @@
-from flask import Flask, request, jsonify, Response,redirect
+import os
+import json
+from flask import Flask, request, jsonify, Response, stream_with_context
 from flask_cors import CORS
-import time
-import random
-from text2generation import generate_long_answer  # Your own function
+import httpx
+
 
 app = Flask(__name__)
 CORS(app)
 
-@app.route('/health')
-def health_check():
+OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_NEW")
+
+@app.route("/health", methods=["GET"])
+def health():
     return jsonify({"status": "ready"}), 200
 
-# Standard answer route
-@app.route('/ask', methods=['POST'])
-def ask_question():
+def openrouter_headers():
+    return {
+        "Authorization": f"Bearer {OPENROUTER_API_KEY}",
+        "Content-Type": "application/json",
+        "HTTP-Referer": "https://sreepathi-ravikumar-sample1.hf.space",
+        "X-Title": "Educational AI Assistant",
+    }
+
+def build_payload(question: str) -> dict:
+    return {
+        "model": "deepseek/deepseek-chat-v3-0324:free",
+        "stream": True,
+        "messages": [{
+            "role": "user",
+            "content": "condition every topics or heading only must startswith # and endswith :\n\n" + question,
+        }],
+        "temperature": 0.3,
+        "max_tokens": 599,
+    }
+
+def stream_openrouter(question: str):
+    with httpx.Client(timeout=None) as client:
+        with client.stream(
+            "POST",
+            "https://openrouter.ai/api/v1/chat/completions",
+            headers=openrouter_headers(),
+            json=build_payload(question),
+        ) as r:
+            r.raise_for_status()
+            # Optional start signal
+            yield "event: start\ndata: stream\n\n"
+            for raw in r.iter_lines():
+                if not raw:
+                    continue
+                # OpenRouter uses SSE-like "data:" lines
+                if raw.startswith(b"data:"):
+                    data = raw[5:].strip()
+                else:
+                    data = raw.strip()
+
+                if data == b"[DONE]":
+                    yield "event: done\ndata: [DONE]\n\n"
+                    break
+
+                try:
+                    obj = json.loads(data.decode("utf-8"))
+                    delta = obj.get("choices", [{}])[0].get("delta", {}).get("content")
+                    if delta:
+                        # Wrap each token in JSON so the frontend can parse safely
+                        yield f"data: {json.dumps({'text': delta})}\n\n"
+                except Exception:
+                    # Fallback: forward as plain text
+                    try:
+                        txt = data.decode("utf-8")
+                    except Exception:
+                        txt = ""
+                    if txt:
+                        yield f"data: {json.dumps({'text': txt})}\n\n"
+
+@app.route("/ask", methods=["POST"])
+def ask():
     try:
-        data = request.get_json()
-        question = data.get('question', '').strip()
+        if not OPENROUTER_API_KEY:
+            return jsonify({"error": "OPENROUTER_API_NEW is not set"}), 500
 
+        data = request.get_json(silent=True) or {}
+        # Accept both keys for compatibility
+        question = (data.get("question") or data.get("userPrompt") or "").strip()
         if not question:
             return jsonify({"error": "Question is required"}), 400
 
-        answer = generate_long_answer(question)
-        return jsonify({"question": question, "answer": answer})
-
+        # Stream back as SSE
+        resp = Response(
+            stream_with_context(stream_openrouter(question)),
+            mimetype="text/event-stream",
+        )
+        # Important headers to avoid buffering by proxies/CDNs
+        resp.headers["Cache-Control"] = "no-cache"
+        resp.headers["X-Accel-Buffering"] = "no"
+        resp.headers["Connection"] = "keep-alive"
+        # CORS for SSE: allow the local frontend origins
+        origin = request.headers.get("Origin")
+        if origin in ALLOWED_ORIGINS:
+            resp.headers["Access-Control-Allow-Origin"] = origin
+            resp.headers["Vary"] = "Origin"
+        return resp
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-@app.route('/get_video')
-def get_video():
-    video_lst=[
-        "https://videos.pexels.com/video-files/3571264/3571264-sd_640_360_30fps.mp4",
-        "https://videos.pexels.com/video-files/3052446/3052446-hd_1920_1080_30fps.mp4",
-        "https://videos.pexels.com/video-files/6394054/6394054-sd_960_480_24fps.mp4",
-        
-    ]
-    return redirect(random.choice(video_lst))
-
-
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+if __name__ == "__main__":
+    # In Spaces, use the default port; locally, 7860 is fine.
+    app.run(host="0.0.0.0", port=7860, debug=True, threaded=True)