Update app.py

app.py

@@ -31,23 +31,67 @@ def get_api_keys():
         st.error(f"Error loading API keys: {str(e)}")
         return None, None
 
+# In app.py
+
 def run_market_research(topic: str, progress_container):
     """Run the market research process"""
     try:
         # Create and run the crew
         crew = create_research_crew(topic)
         
-        # Update progress
-        update_progress(progress_container, 25, "Gathering market data...")
-        update_progress(progress_container, 50, "Analyzing findings...")
-        update_progress(progress_container, 75, "Generating report...")
+        # Update progress for research phase
+        update_progress(progress_container, 25, "Gathering market data and research...")
+        st.write("🔍 Research Analyst is gathering comprehensive market data...")
+        
+        # Update progress for analysis phase
+        update_progress(progress_container, 50, "Analyzing market findings...")
+        st.write("📊 Data Analyst is processing market insights...")
+        
+        # Update progress for report generation
+        update_progress(progress_container, 75, "Compiling comprehensive report...")
+        st.write("✍️ Report Writer is creating the final document...")
         
         # Execute the crew
         result = crew.kickoff()
         
-        # Format and return the 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 structured sections
+        sections = raw_text.split('===')
+        processed_report = {}
+        
+        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('-')
+            ]
+        }
+        
+        # Update final progress
         update_progress(progress_container, 100, "Report completed!")
-        return format_json_output(result)
+        st.success("✨ Market research report generated successfully!")
+        
+        return final_report
 
     except Exception as e:
         st.error(f"Error during research: {str(e)}")