Update utils.py

utils.py

@@ -141,6 +141,9 @@ def process_crew_output(crew_result, topic):
                 'analysis_type': 'Report Writing'
             }
         }
+
+        # Get base report
+        base_report = str(crew_result) if crew_result else "No report generated"
         
         # Extract individual agent outputs from crew_result
         if hasattr(crew_result, 'tasks'):
@@ -151,12 +154,41 @@ def process_crew_output(crew_result, topic):
                     agent_outputs['analyst']['raw_output'] = task.output if task.output else "No analysis output available"
                 elif 'writer' in task.agent.role.lower():
                     agent_outputs['writer']['raw_output'] = task.output if task.output else "No writer output available"
-        
-        # Get base report
-        base_report = str(crew_result)
-        
-        # Rest of the function remains the same...
-        
+
+        # Extract metrics with error handling
+        try:
+            metrics = extract_metrics(base_report)
+        except Exception as e:
+            st.warning(f"Warning extracting metrics: {str(e)}")
+            metrics = {
+                'market_size': 'N/A',
+                'cagr': 'N/A',
+                'leader_share': 'N/A',
+                'key_players': 'N/A',
+                'key_regions': 'N/A',
+                'dominant_segment': 'N/A'
+            }
+
+        # Generate enhanced report content
+        try:
+            enhanced_content = enhance_report_with_gpt4(base_report, topic)
+        except Exception as e:
+            st.warning(f"Warning enhancing report: {str(e)}")
+            enhanced_content = base_report
+
+        # Extract market data
+        try:
+            market_data = extract_market_data(base_report)
+        except Exception as e:
+            st.warning(f"Warning extracting market data: {str(e)}")
+            market_data = {
+                "marketShares": [],
+                "growthTrend": [],
+                "regionalDistribution": [],
+                "techAdoption": [],
+                "keyPlayers": []
+            }
+
         return {
             'metrics': metrics,
             'content': enhanced_content,
@@ -167,12 +199,26 @@ def process_crew_output(crew_result, topic):
         
     except Exception as e:
         st.error(f"Error processing report: {str(e)}")
+        # Return default structure
         return {
-            'metrics': {},
+            'metrics': {
+                'market_size': 'N/A',
+                'cagr': 'N/A',
+                'leader_share': 'N/A',
+                'key_players': 'N/A',
+                'key_regions': 'N/A',
+                'dominant_segment': 'N/A'
+            },
             'content': "Error generating report content",
-            'raw': str(crew_result),
+            'raw': str(crew_result) if crew_result else "No report generated",
             'agent_outputs': agent_outputs,
-            'market_data': {}
+            'market_data': {
+                "marketShares": [],
+                "growthTrend": [],
+                "regionalDistribution": [],
+                "techAdoption": [],
+                "keyPlayers": []
+            }
         }
 
 def extract_market_data(text):