Update app.py

app.py

@@ -105,44 +105,44 @@ if st.button("Generate Prediction and Analysis"):
         f"Be as accurate and specific to Kenya as possible in your analysis. And please do not generate long texts, make it as short and precise as possible, i am stressing on this please, generate something short."
     )
 
-   try:
-    stages = [
-        "Analyzing climate conditions...",
-        "Checking location data...",
-        "Fetching historical data...",
-        "Running simulations...",
-        "Processing current weather...",
-        "Assessing environmental factors...",
-        "Calculating predictions...",
-        "Compiling results...",
-        "Finalizing analysis...",
-        "Preparing output..."
-    ]
-    
-    with st.spinner("Analyzing climate conditions and generating predictions..."):
-        # Loop through each stage, updating the spinner text and waiting for 2 seconds
-        for stage in stages:
-            st.spinner(stage)
-            time.sleep(2)
+        try:
+        stages = [
+            "Analyzing climate conditions...",
+            "Checking location data...",
+            "Fetching historical data...",
+            "Running simulations...",
+            "Processing current weather...",
+            "Assessing environmental factors...",
+            "Calculating predictions...",
+            "Compiling results...",
+            "Finalizing analysis...",
+            "Preparing output..."
+        ]
         
-        initial_response = call_ai_model_initial(all_message)
-
-        initial_text = ""
-        for line in initial_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:
-                            initial_text += delta["content"]
-                except json.JSONDecodeError:
-                    continue
-
-        st.success("Analysis completed!")
+        with st.spinner("Analyzing climate conditions and generating predictions..."):
+            # Loop through each stage, updating the spinner text and waiting for 2 seconds
+            for stage in stages:
+                st.spinner(stage)
+                time.sleep(2)
+            
+            initial_response = call_ai_model_initial(all_message)
+    
+            initial_text = ""
+            for line in initial_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:
+                                initial_text += delta["content"]
+                    except json.JSONDecodeError:
+                        continue
+    
+            st.success("Analysis completed!")
         # Display prediction
         st.subheader("Climate Impact Analysis for Sports in Kenya")
         st.markdown(initial_text.strip())