Update brain.py

brain.py

@@ -23,25 +23,29 @@ model = Llama(
 
 system_prompt = "You are RiShre AI. Be helpful and concise. You are made by Badge94 organization. Never say you are mistral or anyone else"
 
-@app.route("/api/stream", methods=["GET"])
+# Flask Code (brain.py ya app.py)
+
+@app.route("/api/chat", methods=["POST"]) # Route badal kar /api/chat kiya aur POST allow kiya
 def stream():
+    # Frontend se data nikalna
     data = request.get_json()
     user_msg = data.get("message", "")
 
     prompt = f"<s>[INST] {system_prompt}\n\n{user_msg} [/INST]"
 
     def generate():
+        # model() function se streaming chunks nikalna
         for chunk in model(
             prompt,
-            max_tokens=120,
+            max_tokens=256, # Tokens thode badha diye taaki lamba answer aa sake
             temperature=0.7,
-            top_p=0.9,
-            stream=True   # 🔥 KEY LINE
+            stream=True
         ):
             token = chunk["choices"][0]["text"]
             if token:
+                # SSE Format: "data: token_content\n\n"
                 yield f"data: {token}\n\n"
-
+        
         yield "data: [DONE]\n\n"
 
     return Response(generate(), mimetype="text/event-stream")