Update app.py

app.py

@@ -200,91 +200,92 @@ def main():
 
     with tab2:
         if 'current_report' in st.session_state:
-            report = st.session_state.current_report
-            topic = st.session_state.current_topic
+            try:
+                report = st.session_state.current_report
+                topic = st.session_state.current_topic
 
-            st.markdown("""
-                <div class="report-container">
-                    <!-- AI Disclaimer -->
-                    <div class="disclaimer">
-                        ⚠️ This report was generated using AI. While we strive for accuracy, 
-                        please verify critical information independently before making decisions.
-                    </div>
-                    
-                    <!-- Report Title -->
-                    <h1 class="section-heading">📊 Market Research Report: {}</h1>
-                    
-                    <!-- Executive Summary -->
-                    <div class="report-section executive-summary">
-                        <h2 class="section-heading">Executive Summary</h2>
-                        <div class="content-block">{}</div>
-                        
-                        <h3 class="subsection-heading">Key Market Highlights</h3>
-                        <div class="content-block">{}</div>
-                        
-                        <h3 class="subsection-heading">Strategic Implications</h3>
-                        <div class="content-block">{}</div>
+                st.markdown("""
+                    <div class="report-container">
+                        <!-- AI Disclaimer -->
+                        <div class="disclaimer">
+                            ⚠️ This report was generated using AI. While we strive for accuracy, 
+                            please verify critical information independently before making decisions.
+                        </div>
                         
-                        <h3 class="subsection-heading">Recommendations</h3>
-                        <div class="content-block">{}</div>
-                    </div>
-                    
-                    <!-- Market Analysis -->
-                    <div class="report-section market-analysis">
-                        <h2 class="section-heading">Market Analysis</h2>
+                        <!-- Report Title -->
+                        <h1 class="section-heading">📊 Market Research Report: {}</h1>
                         
-                        <h3 class="subsection-heading">Market Overview</h3>
-                        <div class="content-block">{}</div>
+                        <!-- Executive Summary -->
+                        <div class="report-section executive-summary">
+                            <h2 class="section-heading">Executive Summary</h2>
+                            <div class="content-block">{}</div>
+                            
+                            <h3 class="subsection-heading">Key Market Highlights</h3>
+                            <div class="content-block">{}</div>
+                            
+                            <h3 class="subsection-heading">Strategic Implications</h3>
+                            <div class="content-block">{}</div>
+                            
+                            <h3 class="subsection-heading">Recommendations</h3>
+                            <div class="content-block">{}</div>
+                        </div>
                         
-                        <h3 class="subsection-heading">Industry Dynamics</h3>
-                        <div class="content-block">{}</div>
+                        <!-- Market Analysis -->
+                        <div class="report-section market-analysis">
+                            <h2 class="section-heading">Market Analysis</h2>
+                            
+                            <h3 class="subsection-heading">Market Overview</h3>
+                            <div class="content-block">{}</div>
+                            
+                            <h3 class="subsection-heading">Industry Dynamics</h3>
+                            <div class="content-block">{}</div>
+                            
+                            <h3 class="subsection-heading">Competitive Landscape</h3>
+                            <div class="content-block">{}</div>
+                            
+                            <h3 class="subsection-heading">Strategic Analysis</h3>
+                            <div class="content-block">{}</div>
+                        </div>
                         
-                        <h3 class="subsection-heading">Competitive Landscape</h3>
-                        <div class="content-block">{}</div>
+                        <!-- Future Outlook -->
+                        <div class="report-section future-outlook">
+                            <h2 class="section-heading">Future Outlook</h2>
+                            <div class="content-block">{}</div>
+                        </div>
                         
-                        <h3 class="subsection-heading">Strategic Analysis</h3>
-                        <div class="content-block">{}</div>
-                    </div>
-                    
-                    <!-- Future Outlook -->
-                    <div class="report-section future-outlook">
-                        <h2 class="section-heading">Future Outlook</h2>
-                        <div class="content-block">{}</div>
+                        <!-- Sources -->
+                        <div class="report-section sources-section">
+                            <h2 class="section-heading">Sources</h2>
+                            {}
+                        </div>
                     </div>
-                    
-                    <!-- Sources -->
-                    <div class="report-section sources-section">
-                        <h2 class="section-heading">Sources</h2>
-                        {}
+                """.format(
+                    topic,
+                    report['exec_summary']['summary'],
+                    report['exec_summary']['market_highlights'],
+                    report['exec_summary']['strategic_implications'],
+                    report['exec_summary']['recommendations'],
+                    report['market_analysis']['overview'],
+                    report['market_analysis']['dynamics'],
+                    report['market_analysis']['competitive_landscape'],
+                    report['market_analysis']['strategic_analysis'],
+                    report['future_outlook'],
+                    '\n'.join([f'<div class="source-item">• {source}</div>' for source in report['sources']])
+                ), unsafe_allow_html=True)
+                
+                # Download button with custom styling
+                st.markdown("""
+                    <div style="text-align: center; margin-top: 30px;">
+                        <button class="download-button" onclick="alert('Use the Streamlit download button below')">
+                            📥 Download Complete Report
+                        </button>
                     </div>
-                </div>
-            """.format(
-                topic,
-                report['exec_summary']['summary'],
-                report['exec_summary']['market_highlights'],
-                report['exec_summary']['strategic_implications'],
-                report['exec_summary']['recommendations'],
-                report['market_analysis']['overview'],
-                report['market_analysis']['dynamics'],
-                report['market_analysis']['competitive_landscape'],
-                report['market_analysis']['strategic_analysis'],
-                report['future_outlook'],
-                '\n'.join([f'<div class="source-item">• {source}</div>' for source in report['sources']])
-            ), unsafe_allow_html=True)
-            
-            # Download button with custom styling
-            st.markdown("""
-                <div style="text-align: center; margin-top: 30px;">
-                    <button class="download-button" onclick="alert('Use the Streamlit download button below')">
-                        📥 Download Complete Report
-                    </button>
-                </div>
-            """, unsafe_allow_html=True)
-            
-            # Actual download functionality (hidden but functional)
-            st.download_button(
-                "Download Report",
-                data=f"""# Market Research Report: {topic}
+                """, unsafe_allow_html=True)
+                
+                # Actual download functionality
+                st.download_button(
+                    "Download Report",
+                    data=f"""# Market Research Report: {topic}
 
 ## Executive Summary
 {report['exec_summary']['summary']}
@@ -316,17 +317,16 @@ def main():
 {report['future_outlook']}
 
 ## Sources
-{chr(10).join([f"• {source}" for source in report['sources']])}
-""",
-                file_name=f"{topic.lower().replace(' ', '_')}_market_research.md",
-                mime="text/markdown",
-                key='download_button',
-                help="Click to download the complete report",
-            )
+{chr(10).join([f"• {source}" for source in report['sources']])}""",
+                    file_name=f"{topic.lower().replace(' ', '_')}_market_research.md",
+                    mime="text/markdown",
+                    key='download_button',
+                    help="Click to download the complete report"
+                )
+
             except Exception as e:
                 st.error(f"Error displaying report: {str(e)}")
                 st.write("Raw report data:", report)  # Debug information
-            
 
 if __name__ == "__main__":
     main()