Update app.py

app.py

@@ -13,7 +13,7 @@ load_dotenv()
 # Set page config
 st.set_page_config(
     page_title="Market Research Generator",
-    page_icon="📊",
+    page_icon="📈",
     layout="wide",
     initial_sidebar_state="expanded"
 )
@@ -226,8 +226,7 @@ def run_market_research(topic: str, progress_container, chat_container, log_cont
         # Step 2: Gathering market data
         update_progress(progress_container, 25, "Gathering market data...")
         log_agent_activity("Research Analyst", f"Beginning comprehensive research on {topic}")
-        display_agent_message(chat_container, "researcher", 
-            f"Starting comprehensive research on {topic}...")
+        time.sleep(1)
         log_agent_activity("Research Analyst", "Collecting market size data...")
         time.sleep(1)
         log_agent_activity("Research Analyst", "Analyzing competitive landscape...")
@@ -238,8 +237,7 @@ def run_market_research(topic: str, progress_container, chat_container, log_cont
         # Step 3: Analyzing findings
         update_progress(progress_container, 50, "Analyzing findings...")
         log_agent_activity("Data Analyst", "Beginning data analysis phase")
-        display_agent_message(chat_container, "analyst", 
-            "Processing research data and identifying key insights...")
+        time.sleep(1)
         log_agent_activity("Data Analyst", "Processing market size metrics...")
         time.sleep(1)
         log_agent_activity("Data Analyst", "Calculating growth rates...")
@@ -250,8 +248,7 @@ def run_market_research(topic: str, progress_container, chat_container, log_cont
         # Step 4: Compiling final report
         update_progress(progress_container, 75, "Compiling final report...")
         log_agent_activity("Report Writer", "Starting report compilation")
-        display_agent_message(chat_container, "writer", 
-            "Compiling final report with executive summary and detailed analysis...")
+        time.sleep(1)
         log_agent_activity("Report Writer", "Creating executive summary...")
         time.sleep(1)
         log_agent_activity("Report Writer", "Drafting detailed analysis...")
@@ -269,10 +266,6 @@ def run_market_research(topic: str, progress_container, chat_container, log_cont
         # Format and process the result
         report_data = format_json_output(result)
         
-        # Add visualizations
-        charts = generate_visualizations(report_data)
-        report_data['charts'] = charts
-        
         update_progress(progress_container, 100, "Report completed!")
         st.success("✨ Report generation completed! You can now view the full report.")
         log_agent_activity("System", "Report generation completed!")
@@ -445,4 +438,11 @@ def format_json_output(raw_output):
             "metrics": {
                 "market_size_data": [],
                 "growth_rates": [],
-                "market_shares":
+                "market_shares": []
+            }
+        }
+    except Exception as e:
+        st.error(f"Error formatting output: {str(e)}")
+
+if __name__ == "__main__":
+    main()