reverting frontend

app.py

@@ -38,20 +38,21 @@ def process_faq(question, user_id="anonymous", model="claude-sonnet"):
             if response.status_code != 200:
                 return f"Error: Lambda function returned status code {response.status_code}"
             
-            # Process the response (Lambda returns complete response as plain text)
+            # Process the streaming response
             full_response = ""
             for chunk in response.iter_content(chunk_size=1024, decode_unicode=True):
                 if chunk:
-                    full_response += chunk
-            
-            # Simulate streaming by yielding progressively
-            if full_response:
-                words = full_response.split()
-                current_text = ""
-                for i, word in enumerate(words):
-                    current_text += word + " "
-                    if i % 2 == 0 or i == len(words) - 1:  # Yield every 2 words or at the end
-                        yield current_text.strip()
+                    try:
+                        # Try to parse the chunk as JSON
+                        chunk_data = json.loads(chunk)
+                        if "response" in chunk_data:
+                            chunk_text = chunk_data["response"]
+                            full_response += chunk_text
+                            yield full_response
+                    except json.JSONDecodeError:
+                        # If not JSON, treat as plain text
+                        full_response += chunk
+                        yield full_response
             
             return full_response