Update app.py

app.py

@@ -103,6 +103,7 @@ def main():
                     finally:
                         st.session_state.generating = False
     
+    # Inside the tab2 section of main()
     with tab2:
         if 'current_report' in st.session_state:
             report = st.session_state.current_report
@@ -111,36 +112,56 @@ def main():
             # Display AI Disclaimer
             st.info("⚠️ This report was generated using AI. Please verify critical information independently.")
             
-            # Executive Summary
-            st.subheader("Executive Summary")
-            st.write(report['exec_summary']['summary'])
-            
-            # Key Metrics
-            col1, col2, col3 = st.columns(3)
-            with col1:
-                st.metric("Market Size", report['exec_summary']['market_size'])
-            with col2:
-                st.metric("Growth Rate", report['exec_summary']['growth_rate'])
-            with col3:
-                st.metric("Key Players", report['exec_summary']['key_players'])
-            
-            # Detailed Report
-            st.subheader("Detailed Analysis")
-            st.write(report['detailed_report'])
-            
-            # Sources
-            if report['sources']:
-                st.subheader("Sources")
-                for source in report['sources']:
-                    st.markdown(f"- {source}")
-            
-            # Download button
-            st.download_button(
-                "Download Report",
-                data=str(report['detailed_report']),
-                file_name=f"{topic.lower().replace(' ', '_')}_market_research.md",
-                mime="text/markdown"
-            )
+            try:
+                # Executive Summary Section with error handling
+                st.subheader("Executive Summary")
+                if report.get('exec_summary', {}).get('summary'):
+                    st.write(report['exec_summary']['summary'])
+                else:
+                    st.write("Executive summary not available")
+
+                # Key Metrics with error handling
+                col1, col2, col3 = st.columns(3)
+                with col1:
+                    market_size = report.get('exec_summary', {}).get('market_size', 'N/A')
+                    st.metric("Market Size", market_size)
+                with col2:
+                    growth_rate = report.get('exec_summary', {}).get('growth_rate', 'N/A')
+                    st.metric("Growth Rate", growth_rate)
+                with col3:
+                    key_players = report.get('exec_summary', {}).get('key_players', 'N/A')
+                    st.metric("Key Players", key_players)
+
+                # Detailed Report Section
+                st.subheader("Detailed Analysis")
+                detailed_report = report.get('detailed_report', 'Detailed analysis not available')
+                st.write(detailed_report)
+
+                # Sources Section
+                sources = report.get('sources', [])
+                if sources:
+                    st.subheader("Sources")
+                    for source in sources:
+                        st.markdown(f"- {source}")
 
+                # Download button
+                report_text = (
+                    f"# Market Research Report: {topic}\n\n"
+                    f"## Executive Summary\n{report['exec_summary']['summary']}\n\n"
+                    f"### Key Metrics\n"
+                    f"- Market Size: {report['exec_summary']['market_size']}\n"
+                    f"- Growth Rate: {report['exec_summary']['growth_rate']}\n"
+                    f"- Key Players: {report['exec_summary']['key_players']}\n\n"
+                    f"## Detailed Analysis\n{report['detailed_report']}\n\n"
+                    f"## Sources\n" + "\n".join([f"- {source}" for source in report['sources']])
+                )
+                
+                st.download_button(
+                    "Download Report",
+                    data=report_text,
+                    file_name=f"{topic.lower().replace(' ', '_')}_market_research.md",
+                    mime="text/markdown"
+                )
+                
 if __name__ == "__main__":
     main()