Update app.py

app.py

@@ -29,10 +29,7 @@ def call_ai_model(all_message):
     response = requests.post(url, json=payload, headers=headers, stream=True)
     response.raise_for_status()  # Ensure HTTP request was successful
 
-    # Debugging line to print raw response
-    for line in response.iter_lines():
-        if line:
-            yield line.decode('utf-8')
+    return response
 
 # Streamlit app layout
 st.title("Climate Change Impact on Sports Using AI")
@@ -49,8 +46,25 @@ if st.button("Generate Prediction"):
     
     try:
         with st.spinner("Generating response..."):
-            response_lines = call_ai_model(all_message)
-            generated_text = "".join(response_lines)
+            response = call_ai_model(all_message)
+            
+            # Handle the streaming response and concatenate chunks
+            generated_text = ""
+            for line in response.iter_lines():
+                if line:
+                    line_content = line.decode('utf-8')
+                    if line_content.startswith("data: "):
+                        line_content = line_content[6:]  # Strip "data: " prefix
+                    try:
+                        json_data = json.loads(line_content)
+                        if "choices" in json_data:
+                            delta = json_data["choices"][0]["delta"]
+                            if "content" in delta:
+                                generated_text += delta["content"]
+                                st.write(generated_text)  # Update the displayed text incrementally
+                    except json.JSONDecodeError:
+                        continue
+            
             st.success("Response generated successfully!")
             st.write(generated_text)
     except ValueError as ve: