Update main.py

main.py

@@ -58,19 +58,19 @@ def generate_search(query: str, stream: bool = True) -> str:
     # Collect streamed text content
     streaming_text = ""
     for value in response.iter_lines(decode_unicode=True):
-        # Ensure the value is clean and processable
-        if value.startswith("data: "):  # Ensure the prefix matches expected format
+        # Ensure the value starts with 'data: ' and process it
+        if value.startswith("data: "):  
             try:
                 json_modified_value = json.loads(value[6:])  # Remove 'data: ' prefix
                 content = json_modified_value.get("choices", [{}])[0].get("delta", {}).get("content", "")
-                if content.strip():  # Avoid empty content from delta
-                    if stream:
-                        yield f"data: {json.dumps({'response': content})}\n\n"
-                    streaming_text += content
+                # Include everything, even if it's just whitespace
+                if stream:
+                    yield f"data: {json.dumps({'response': content})}\n\n"
+                streaming_text += content
             except json.JSONDecodeError:
                 continue  # Skip lines that are not valid JSON
     
-    # If not streaming, yield the full text
+    # If not streaming, yield the full collected content
     if not stream:
         yield streaming_text