Update app.py

app.py

@@ -56,28 +56,28 @@ def run_market_research(topic: str, progress_container):
 
 def main():
     st.title("🤖 AI Market Research Generator")
-    
+
     # Get API keys
     serper_key, openai_key = get_api_keys()
-    
+
     if not serper_key or not openai_key:
         st.error("Failed to load required API keys. Please check your Hugging Face Space secrets.")
         return
-    
+
     # Initialize session states
     if 'generating' not in st.session_state:
         st.session_state.generating = False
-    
+
     # Create tabs
     tab1, tab2 = st.tabs(["Generate Report", "View Reports"])
-    
+
     with tab1:
         st.subheader("Enter Research Topic")
         topic = st.text_input(
             "What market would you like to research?",
             placeholder="e.g., Electric Vehicles Market"
         )
-        
+
         if st.session_state.generating:
             st.button("Generating Report...", disabled=True)
         else:
@@ -86,79 +86,142 @@ def main():
                     st.error("Please enter a research topic")
                 else:
                     st.session_state.generating = True
-                    
+
                     # Create progress container
                     progress_container = st.container()
-                    
+
                     try:
                         with st.spinner("Generating comprehensive market research report..."):
                             result = run_market_research(topic, progress_container)
-                            
-                            if result:
-                                st.session_state.current_report = result
-                                st.session_state.current_topic = topic
-                                st.success("Report generated successfully! View it in the Reports tab.")
+
+                        if result:
+                            st.session_state.current_report = result
+                            st.session_state.current_topic = topic
+                            st.success("Report generated successfully! View it in the Reports tab.")
                     except Exception as e:
                         st.error(f"Error generating report: {str(e)}")
                     finally:
                         st.session_state.generating = False
-    
+
     with tab2:
         if 'current_report' in st.session_state:
             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
-                st.header("Executive Summary")
-                st.write(report.get('exec_summary', {}).get('summary', 'No summary available'))
-                
-                # Market Overview
-                st.header("Market Overview")
-                st.write(f"Market Size: {report.get('exec_summary', {}).get('market_size', 'Not available')}")
-                st.write(f"Growth Rate: {report.get('exec_summary', {}).get('growth_rate', 'Not available')}")
-                st.write(f"Key Players: {report.get('exec_summary', {}).get('key_players', 'Not available')}")
-
-                # Detailed Analysis
-                st.header("Detailed Analysis")
-                st.write(report.get('detailed_report', 'No detailed analysis available'))
-
-                # Sources
-                st.header("Sources")
-                sources = report.get('sources', ['No sources available'])
-                for source in sources:
-                    st.write(f"• {source}")
-
-                # Simple download button for the report
-                report_text = f"""
-Market Research Report: {topic}
-
-Executive Summary:
-{report.get('exec_summary', {}).get('summary', 'No summary available')}
-
-Market Overview:
-• Market Size: {report.get('exec_summary', {}).get('market_size', 'Not available')}
-• Growth Rate: {report.get('exec_summary', {}).get('growth_rate', 'Not available')}
-• Key Players: {report.get('exec_summary', {}).get('key_players', 'Not available')}
-
-Detailed Analysis:
-{report.get('detailed_report', 'No detailed analysis available')}
-
-Sources:
-{chr(10).join([f"• {source}" for source in sources])}
-"""
-                
+                # 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 Report",
-                    data=report_text,
-                    file_name=f"{topic}_market_research.txt",
-                    mime="text/plain"
+                    "Download Complete Report",
+                    data=f"""
+# Market Research Report: {topic}
+
+## Executive Summary
+{report.get('exec_summary', {}).get('summary', '')}
+
+### Market Highlights
+{report.get('exec_summary', {}).get('market_highlights', '')}
+
+### Strategic Implications
+{report.get('exec_summary', {}).get('strategic_implications', '')}
+
+### Key Recommendations
+{report.get('exec_summary', {}).get('recommendations', '')}
+
+## Detailed Market Analysis
+
+### Market Overview
+{report.get('market_analysis', {}).get('overview', '')}
+
+### Industry Dynamics
+{report.get('market_analysis', {}).get('dynamics', '')}
+
+### Competitive Landscape
+{report.get('market_analysis', {}).get('competitive_landscape', '')}
+
+### Strategic Analysis
+{report.get('market_analysis', {}).get('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')}
+
+## Sources
+{chr(10).join([f"• {source}" for source in report.get('sources', [])])}
+
+## Appendices
+{report.get('appendices', '')}
+""",
+                    file_name=f"{topic.lower().replace(' ', '_')}_market_research.md",
+                    mime="text/markdown"
                 )
-                
+
             except Exception as e:
                 st.error(f"Error displaying report: {str(e)}")
+                st.write("Raw report data:", report)  # Debug information
+
 if __name__ == "__main__":
     main()