Update app.py

app.py

@@ -103,7 +103,6 @@ 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
@@ -112,7 +111,39 @@ def main():
             # Display AI Disclaimer
             st.info("⚠️ This report was generated using AI. Please verify critical information independently.")
             
-            # Download button with error handling
+            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
                 try:
                     # Safely get report sections with defaults
                     exec_summary = report.get('exec_summary', {})
@@ -141,6 +172,8 @@ def main():
                     )
                 except Exception as e:
                     st.error(f"Error preparing download: {str(e)}")
+            except Exception as e:
+                st.error(f"Error displaying report: {str(e)}")
 
 if __name__ == "__main__":
     main()