Update app.py

app.py

@@ -39,44 +39,49 @@ def run_market_research(topic: str, progress_container):
         # Create and run the crew
         crew = create_research_crew(topic)
         
-        # Update progress for research phase
-        update_progress(progress_container, 25, "Gathering market data and research...")
+        # Update progress and show what's happening
+        update_progress(progress_container, 25, "Gathering market data...")
         st.write("🔍 Research Analyst is gathering comprehensive market data...")
         
-        # Update progress for analysis phase
-        update_progress(progress_container, 50, "Analyzing market findings...")
+        update_progress(progress_container, 50, "Analyzing findings...")
         st.write("📊 Data Analyst is processing market insights...")
         
-        # Update progress for report generation
-        update_progress(progress_container, 75, "Compiling comprehensive report...")
+        update_progress(progress_container, 75, "Generating report...")
         st.write("✍️ Report Writer is creating the final document...")
         
-        # Execute the crew
+        # Execute the crew and get result
         result = crew.kickoff()
         
+        # Debug print
+        st.write("Debug - Raw Result:", result)
+        
         # Get raw text from result
         if hasattr(result, 'raw_output'):
             raw_text = str(result.raw_output)
         else:
             raw_text = str(result)
         
-        # Process the raw text into sections
+        # Debug print
+        st.write("Debug - Raw Text:", raw_text[:500])  # Show first 500 chars
+        
+        # Split the text into sections
         sections = raw_text.split('===')
+        
+        # Initialize the processed report structure
         processed_report = {
             'exec_summary': {
-                'summary': '',
-                'market_highlights': '',
-                'strategic_implications': '',
-                'recommendations': ''
+                'summary': "Executive summary not available",
+                'market_highlights': "Highlights not available",
+                'strategic_implications': "Implications not available",
+                'recommendations': "Recommendations not available"
             },
             'market_analysis': {
-                'overview': '',
-                'dynamics': '',
-                'competitive_landscape': '',
-                'strategic_analysis': ''
+                'overview': "Overview not available",
+                'dynamics': "Industry dynamics not available",
+                'competitive_landscape': "Competitive analysis not available",
+                'strategic_analysis': "Strategic analysis not available"
             },
-            'future_outlook': '',
-            'appendices': '',
+            'future_outlook': "Future outlook not available",
             'sources': []
         }
         
@@ -85,70 +90,59 @@ def run_market_research(topic: str, progress_container):
             if not section.strip():
                 continue
             
+            # Split section into title and content
             parts = section.strip().split('\n', 1)
             if len(parts) == 2:
-                section_name, content = parts
-                section_name = section_name.strip().upper()
-                content = content.strip()
+                section_name = parts[0].strip().upper()
+                content = parts[1].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 ''
+                # Debug print
+                st.write(f"Debug - Processing section: {section_name}")
                 
-                elif section_name == 'MARKET ANALYSIS':
-                    # Process market analysis section
+                if "EXECUTIVE SUMMARY" in section_name:
                     subsections = content.split('##')
-                    processed_report['market_analysis']['overview'] = subsections[0]
+                    processed_report['exec_summary']['summary'] = subsections[0].strip()
+                    
                     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
+                        if 'Market Highlights' in subsection:
+                            processed_report['exec_summary']['market_highlights'] = subsection.split('\n', 1)[1].strip()
+                        elif 'Strategic Implications' in subsection:
+                            processed_report['exec_summary']['strategic_implications'] = subsection.split('\n', 1)[1].strip()
+                        elif 'Recommendations' in subsection:
+                            processed_report['exec_summary']['recommendations'] = subsection.split('\n', 1)[1].strip()
                 
-                elif section_name == 'FUTURE OUTLOOK':
-                    processed_report['future_outlook'] = content
+                elif "MARKET ANALYSIS" in section_name:
+                    subsections = content.split('##')
+                    for subsection in subsections:
+                        if 'Market Overview' in subsection:
+                            processed_report['market_analysis']['overview'] = subsection.split('\n', 1)[1].strip()
+                        elif 'Industry Dynamics' in subsection:
+                            processed_report['market_analysis']['dynamics'] = subsection.split('\n', 1)[1].strip()
+                        elif 'Competitive Landscape' in subsection:
+                            processed_report['market_analysis']['competitive_landscape'] = subsection.split('\n', 1)[1].strip()
+                        elif 'Strategic Analysis' in subsection:
+                            processed_report['market_analysis']['strategic_analysis'] = subsection.split('\n', 1)[1].strip()
                 
-                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('•')]
+                elif "FUTURE OUTLOOK" in section_name:
+                    processed_report['future_outlook'] = content.strip()
+                
+                elif "SOURCES" in section_name or "APPENDICES" in section_name:
+                    sources = [line.strip() for line in content.split('\n') 
+                             if line.strip() and (line.strip().startswith('-') or 
+                                                line.strip().startswith('•'))]
                     processed_report['sources'] = sources
+
+        # Debug print
+        st.write("Debug - Processed Report Structure:", processed_report.keys())
         
-        # Update final progress
         update_progress(progress_container, 100, "Report completed!")
-        st.success("✨ Market research report generated successfully!")
-        
         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 {
-            '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': []
-        }
+        return None
+        
 def main():
     st.title("🤖 AI Market Research Generator")