Update app.py

app.py

@@ -112,56 +112,35 @@ def main():
             # Display AI Disclaimer
             st.info("⚠️ This report was generated using AI. Please verify critical information independently.")
             
-            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")
+            # Download button with error handling
+                try:
+                    # Safely get report sections with defaults
+                    exec_summary = report.get('exec_summary', {})
+                    report_text = f"""# Market Research Report: {topic}
+
+## Executive Summary
+{exec_summary.get('summary', 'No summary 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)
+### Key Metrics
+- Market Size: {exec_summary.get('market_size', 'Not available')}
+- Growth Rate: {exec_summary.get('growth_rate', 'Not available')}
+- Key Players: {exec_summary.get('key_players', 'Not available')}
 
-                # Detailed Report Section
-                st.subheader("Detailed Analysis")
-                detailed_report = report.get('detailed_report', 'Detailed analysis not available')
-                st.write(detailed_report)
+## Detailed Analysis
+{report.get('detailed_report', 'No detailed analysis available')}
 
-                # Sources Section
-                sources = report.get('sources', [])
-                if sources:
-                    st.subheader("Sources")
-                    for source in sources:
-                        st.markdown(f"- {source}")
+## Sources
+{chr(10).join([f"- {source}" for source in report.get('sources', ['No sources available'])])}
+"""
+                    
+                    st.download_button(
+                        label="Download Report",
+                        data=report_text,
+                        file_name=f"{topic.lower().replace(' ', '_')}_market_research.md",
+                        mime="text/markdown"
+                    )
+                except Exception as e:
+                    st.error(f"Error preparing download: {str(e)}")
 
-                # 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()