Refactor chat stream endpoint and update HTML for query parameter handling

- Changed `/chat/stream` endpoint from POST to GET and updated user message retrieval to support query parameters.
- Improved error handling for missing user messages by checking session history.
- Updated `chat.html` to encode user messages for URL query parameters and streamline EventSource creation.
- Enhanced user experience by ensuring proper message handling and session updates during streaming.

Author: TMHSDigital

src/deepseek_wrapper/templates/chat.html

@@ -54,14 +54,15 @@
         if (btn) btn.disabled = true;
         if (sendingStatus) sendingStatus.style.display = 'block';
 
-        const formData = new FormData();
-        formData.append('user_message', input.value);
-        
-        const userMessage = input.value;
+        const userMessageValue = input.value; // Capture value before clearing
         input.value = '';
         autoResizeTextarea(input); // Reset height after clearing
 
-        const eventSource = new EventSource(`/chat/stream?_=${Date.now()}`, {
+        // Encode user_message for URL query parameter
+        const encodedUserMessage = encodeURIComponent(userMessageValue);
+
+        // Create and handle the EventSource for streaming
+        const eventSource = new EventSource(`/chat/stream?user_message=${encodedUserMessage}&_=${Date.now()}`, {
             withCredentials: true
         });
 

src/deepseek_wrapper/web.py

@@ -115,61 +115,48 @@ async def upload_file(file: UploadFile = File(...), request: Request = None):
         request.session["history"] = history
     return JSONResponse({"filename": file.filename, "content_type": file.content_type})
 
-@app.post("/chat/stream")
-async def chat_stream(request: Request):
+@app.get("/chat/stream")
+async def chat_stream(request: Request, user_message: str = None):
     """Stream chat responses using Server-Sent Events."""
     try:
-        # Get form data from request
-        form_data = await request.form()
-        user_message = form_data.get("user_message")
-        
         if not user_message:
-            return JSONResponse({"error": "No message provided"}, status_code=400)
+            history_check = request.session.get("history", [])
+            if history_check and history_check[-1]["role"] == "user":
+                user_message = history_check[-1]["content"]
+            else:
+                return JSONResponse({"error": "No message provided in query parameters"}, status_code=400)
 
-        # Get history from session
         history = request.session.get("history", [])
 
-        # Add user message to history
         user_msg = {"role": "user", "content": user_message, "timestamp": now_str()}
         history.append(user_msg)
 
-        # Update session with user message
         request.session["history"] = history
 
-        # Create an event stream generator
         async def event_generator():
-            # Send user message confirmation first
             yield f"data: {json.dumps({'type': 'user_msg_received', 'message': user_msg})}\n\n"
-            await asyncio.sleep(0.1)  # Small delay to allow frontend rendering
+            await asyncio.sleep(0.1)
 
-            # Start assistant message
             assistant_msg = {"role": "assistant", "content": "", "timestamp": now_str()}
             yield f"data: {json.dumps({'type': 'assistant_msg_start', 'message': assistant_msg})}\n\n"
 
             try:
-                # Start streaming response
                 collected_content = []
                 async for content_chunk in client.async_chat_completion_stream(history):
-                    # Add chunk to the collected content
                     collected_content.append(content_chunk)
                     assistant_msg["content"] = ''.join(collected_content)
 
-                    # Send the chunk to the client
                     yield f"data: {json.dumps({'type': 'content_chunk', 'chunk': content_chunk})}\n\n"
 
-                # Stream is complete - update the history in the session
                 complete_msg = {"role": "assistant", "content": ''.join(collected_content), "timestamp": now_str()}
                 history.append(complete_msg)
                 request.session["history"] = history
 
-                # Send completion message
                 yield f"data: {json.dumps({'type': 'complete', 'message': complete_msg})}\n\n"
             except Exception as e:
-                # Handle errors during streaming
                 error_msg = str(e)
                 yield f"data: {json.dumps({'type': 'error', 'error': error_msg})}\n\n"
 
-                # Add error message to history
                 error_response = f"Error: {error_msg}"
                 history.append({"role": "assistant", "content": error_response, "timestamp": now_str()})
                 request.session["history"] = history