add next quiz to json

main.py

@@ -1,3 +1,4 @@
+import json
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
@@ -62,20 +63,28 @@ async def chat_endpoint(request: QueryRequest):
 # ============================================
 @app.post("/chat-stream")
 async def chat_stream_endpoint(request: QueryRequest):
-    """Streaming endpoint - returns chunks in real-time"""
+    """Streaming endpoint that returns chunks in real-time with a consistent structure"""
     try:
         groq_client = get_groq_client()
 
         def generate():
             try:
+                response = ""
                 for chunk in groq_client.ask_stream(request.query):
-                    yield f"data: {chunk}\n\n"
+                    response += chunk
 
-                # Send completion signal
-                yield "data: [DONE]\n\n"
+                    yield f"data: {json.dumps({'status_code': 200, 'response': chunk, 'next_questions': []})}\n\n"
+
+                if response.startswith("Error while"):
+                    yield f"data: {json.dumps({'status_code': 500, 'response': response, 'next_questions': []})}\n\n"
+                else:
+
+                    yield f"data: {json.dumps({'status_code': 200, 'response': response, 'next_questions': []})}\n\n"
+
+                    yield f"data: {json.dumps({'status_code': 200, 'response': '[DONE]', 'next_questions': groq_client.get_next_questions(request.ip)})}\n\n"
 
             except Exception as e:
-                yield f"data: Error: {str(e)}\n\n"
+                yield f"data: {json.dumps({'status_code': 500, 'response': f'Error: {str(e)}', 'next_questions': []})}\n\n"
 
         return StreamingResponse(
             generate(),
@@ -88,7 +97,11 @@ async def chat_stream_endpoint(request: QueryRequest):
         )
 
     except Exception as e:
-        return {"status_code": 500, "response": f"Internal error: {e}"}
+        return {
+            "status_code": 500,
+            "response": f"Internal error: {e}",
+            "next_questions": [],
+        }
 
 
 @app.get("/")