Update app.py

app.py

@@ -60,44 +60,95 @@ def run_market_research(topic: str, progress_container):
         else:
             raw_text = str(result)
         
-        # Process the raw text into structured sections
+        # Process the raw text into sections
         sections = raw_text.split('===')
-        processed_report = {}
+        processed_report = {
+            'exec_summary': {
+                'summary': '',
+                'market_highlights': '',
+                'strategic_implications': '',
+                'recommendations': ''
+            },
+            'market_analysis': {
+                'overview': '',
+                'dynamics': '',
+                'competitive_landscape': '',
+                'strategic_analysis': ''
+            },
+            'future_outlook': '',
+            'appendices': '',
+            'sources': []
+        }
         
+        # Process each section
         for section in sections:
             if not section.strip():
                 continue
             
-            # Split into section name and content
             parts = section.strip().split('\n', 1)
             if len(parts) == 2:
                 section_name, content = parts
-                processed_report[section_name.strip()] = content.strip()
-        
-        # Create final report structure
-        final_report = {
-            'exec_summary': processed_report.get('EXECUTIVE SUMMARY', ''),
-            'market_analysis': processed_report.get('MARKET ANALYSIS', ''),
-            'future_outlook': processed_report.get('FUTURE OUTLOOK', ''),
-            'appendices': processed_report.get('APPENDICES', ''),
-            'sources': [
-                source.strip() 
-                for source in processed_report.get('APPENDICES', '').split('\n') 
-                if source.strip().startswith('-')
-            ]
-        }
+                section_name = section_name.strip().upper()
+                content = content.strip()
+                
+                if section_name == 'EXECUTIVE SUMMARY':
+                    # Process executive summary
+                    processed_report['exec_summary']['summary'] = content
+                    # Extract main points for highlights
+                    if '##' in content:
+                        highlights = content.split('##')[1:]
+                        processed_report['exec_summary']['market_highlights'] = highlights[0] if highlights else ''
+                        processed_report['exec_summary']['strategic_implications'] = highlights[1] if len(highlights) > 1 else ''
+                        processed_report['exec_summary']['recommendations'] = highlights[2] if len(highlights) > 2 else ''
+                
+                elif section_name == 'MARKET ANALYSIS':
+                    # Process market analysis section
+                    subsections = content.split('##')
+                    processed_report['market_analysis']['overview'] = subsections[0]
+                    for subsection in subsections[1:]:
+                        if 'dynamics' in subsection.lower():
+                            processed_report['market_analysis']['dynamics'] = subsection
+                        elif 'competitive' in subsection.lower():
+                            processed_report['market_analysis']['competitive_landscape'] = subsection
+                        elif 'strategic' in subsection.lower():
+                            processed_report['market_analysis']['strategic_analysis'] = subsection
+                
+                elif section_name == 'FUTURE OUTLOOK':
+                    processed_report['future_outlook'] = content
+                
+                elif section_name == 'APPENDICES':
+                    processed_report['appendices'] = content
+                    # Extract sources from appendices
+                    sources = [line.strip() for line in content.split('\n')
+                             if line.strip().startswith('-') or line.strip().startswith('•')]
+                    processed_report['sources'] = sources
         
         # Update final progress
         update_progress(progress_container, 100, "Report completed!")
         st.success("✨ Market research report generated successfully!")
         
-        return final_report
+        return processed_report
 
     except Exception as e:
         st.error(f"Error during research: {str(e)}")
         print(f"Full error details: {str(e)}")  # For debugging
-        return None
-
+        return {
+            'exec_summary': {
+                'summary': 'Error processing report',
+                'market_highlights': '',
+                'strategic_implications': '',
+                'recommendations': ''
+            },
+            'market_analysis': {
+                'overview': str(raw_text) if 'raw_text' in locals() else 'Error processing analysis',
+                'dynamics': '',
+                'competitive_landscape': '',
+                'strategic_analysis': ''
+            },
+            'future_outlook': '',
+            'appendices': '',
+            'sources': []
+        }
 def main():
     st.title("🤖 AI Market Research Generator")
 
@@ -152,120 +203,130 @@ def main():
             report = st.session_state.current_report
             topic = st.session_state.current_topic
 
-            # Display AI Disclaimer
-            st.info("⚠️ This report was generated using AI. Please verify critical information independently.")
-
-            try:
-                # Executive Summary Section
-                st.header(f"📊 Market Research Report: {topic}")
-
-                with st.expander("Executive Summary", expanded=True):
-                    st.subheader("Key Findings")
-                    st.write(report.get('exec_summary', {}).get('summary', ''))
-
-                    st.subheader("Market Highlights")
-                    st.write(report.get('exec_summary', {}).get('market_highlights', ''))
-
-                    st.subheader("Strategic Implications")
-                    st.write(report.get('exec_summary', {}).get('strategic_implications', ''))
-
-                    st.subheader("Key Recommendations")
-                    st.write(report.get('exec_summary', {}).get('recommendations', ''))
-
-                # Market Analysis Section
-                with st.expander("Detailed Market Analysis", expanded=False):
-                    st.subheader("Market Overview")
-                    st.write(report.get('market_analysis', {}).get('overview', ''))
-
-                    st.subheader("Industry Dynamics")
-                    st.write(report.get('market_analysis', {}).get('dynamics', ''))
-
-                    st.subheader("Competitive Landscape")
-                    st.write(report.get('market_analysis', {}).get('competitive_landscape', ''))
-
-                    st.subheader("Strategic Analysis")
-                    st.write(report.get('market_analysis', {}).get('strategic_analysis', ''))
-
-                # Future Outlook Section
-                with st.expander("Future Outlook", expanded=False):
-                    st.write(report.get('future_outlook', ''))
-
-                # Key Metrics Section
-                with st.expander("Key Metrics", expanded=False):
-                    metrics = report.get('metrics', {})
-                    col1, col2 = st.columns(2)
-
-                    with col1:
-                        st.metric("Market Size", metrics.get('market_size', 'N/A'))
-                        st.metric("Growth Rate", metrics.get('growth_rate', 'N/A'))
-
-                    with col2:
-                        st.metric("Market Share", metrics.get('market_shares', 'N/A'))
-                        st.metric("Other Metrics", metrics.get('key_metrics', 'N/A'))
-
-                # Sources Section
-                with st.expander("Sources & Appendices", expanded=False):
-                    st.subheader("Sources")
-                    sources = report.get('sources', [])
-                    for source in sources:
-                        st.markdown(f"• {source}")
-
-                    st.subheader("Appendices")
-                    st.write(report.get('appendices', ''))
-
-                # Download Full Report
-                st.download_button(
-                    "Download Complete Report",
-                    data=f"""
-# Market Research Report: {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>
+                        
+                        <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>
+                        
+                        <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>
+                    
+                    <!-- Future Outlook -->
+                    <div class="report-section future-outlook">
+                        <h2 class="section-heading">Future Outlook</h2>
+                        <div class="content-block">{}</div>
+                    </div>
+                    
+                    <!-- Sources -->
+                    <div class="report-section sources-section">
+                        <h2 class="section-heading">Sources</h2>
+                        {}
+                    </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}
 
 ## Executive Summary
-{report.get('exec_summary', {}).get('summary', '')}
+{report['exec_summary']['summary']}
 
 ### Market Highlights
-{report.get('exec_summary', {}).get('market_highlights', '')}
+{report['exec_summary']['market_highlights']}
 
 ### Strategic Implications
-{report.get('exec_summary', {}).get('strategic_implications', '')}
+{report['exec_summary']['strategic_implications']}
 
 ### Key Recommendations
-{report.get('exec_summary', {}).get('recommendations', '')}
+{report['exec_summary']['recommendations']}
 
-## Detailed Market Analysis
+## Market Analysis
 
 ### Market Overview
-{report.get('market_analysis', {}).get('overview', '')}
+{report['market_analysis']['overview']}
 
 ### Industry Dynamics
-{report.get('market_analysis', {}).get('dynamics', '')}
+{report['market_analysis']['dynamics']}
 
 ### Competitive Landscape
-{report.get('market_analysis', {}).get('competitive_landscape', '')}
+{report['market_analysis']['competitive_landscape']}
 
 ### Strategic Analysis
-{report.get('market_analysis', {}).get('strategic_analysis', '')}
+{report['market_analysis']['strategic_analysis']}
 
 ## Future Outlook
-{report.get('future_outlook', '')}
-
-## Key Metrics
-- Market Size: {report.get('metrics', {}).get('market_size', 'N/A')}
-- Growth Rate: {report.get('metrics', {}).get('growth_rate', 'N/A')}
-- Market Share: {report.get('metrics', {}).get('market_shares', 'N/A')}
+{report['future_outlook']}
 
 ## Sources
-{chr(10).join([f"• {source}" for source in report.get('sources', [])])}
-
-## Appendices
-{report.get('appendices', '')}
+{chr(10).join([f"• {source}" for source in report['sources']])}
 """,
-                    file_name=f"{topic.lower().replace(' ', '_')}_market_research.md",
-                    mime="text/markdown"
-                )
-
+                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()