Update app.py

app.py

@@ -54,22 +54,103 @@ class FastFinancialAnalyzer:
             return 0.0
 
     def extract_key_metrics(self, income_data, balance_data):
-        """Extract key financial metrics"""
+    """Extract key financial metrics with safety checks"""
+    try:
+        # First, safely extract values with error handling
+        revenue_2025 = self.safe_extract_number(income_data, "Total Net Revenue", "2025")
+        revenue_2021 = self.safe_extract_number(income_data, "Total Net Revenue", "2021")
+        profit_2025 = self.safe_extract_number(income_data, "Net Earnings", "2025")
+        profit_2021 = self.safe_extract_number(income_data, "Net Earnings", "2021")
+        assets_2025 = self.safe_extract_number(balance_data, "Total_Assets", "2025")
+        assets_2021 = self.safe_extract_number(balance_data, "Total_Assets", "2021")
+
         metrics = {
             "Revenue": {
-                "2025": self.clean_number(income_data.get("Total Net Revenue", {}).get("2025", "0")),
-                "2021": self.clean_number(income_data.get("Total Net Revenue", {}).get("2021", "0"))
+                "2025": revenue_2025,
+                "2021": revenue_2021,
+                "Growth": self.calculate_growth(revenue_2025, revenue_2021)
             },
             "Profit": {
-                "2025": self.clean_number(income_data.get("Net Earnings", {}).get("2025", "0")),
-                "2021": self.clean_number(income_data.get("Net Earnings", {}).get("2021", "0"))
+                "2025": profit_2025,
+                "2021": profit_2021,
+                "Growth": self.calculate_growth(profit_2025, profit_2021),
+                "Margin_2025": self.calculate_percentage(profit_2025, revenue_2025)
             },
             "Assets": {
-                "2025": self.clean_number(balance_data.get("Total Assets", {}).get("2025", "0")),
-                "2021": self.clean_number(balance_data.get("Total Assets", {}).get("2021", "0"))
+                "2025": assets_2025,
+                "2021": assets_2021,
+                "Growth": self.calculate_growth(assets_2025, assets_2021)
             }
         }
         return metrics
+    except Exception as e:
+        print(f"Error in metric extraction: {str(e)}")
+        return self.get_default_metrics()
+
+    def safe_extract_number(self, data_dict, key, year):
+    """Safely extract and convert number from data"""
+        try:
+        if isinstance(data_dict, dict):
+            for k, v in data_dict.items():
+                if isinstance(v, dict) and key in k:
+                    value = v.get(year, '0')
+                    return self.clean_number(value)
+            return 0.0
+        except Exception as e:
+            print(f"Error extracting {key} for {year}: {str(e)}")
+            return 0.0
+
+    def calculate_growth(self, current, previous):
+    """Calculate growth percentage with safety check"""
+        try:
+            if previous and previous != 0:
+                return ((current - previous) / abs(previous)) * 100
+            return 0.0
+        except:
+            return 0.0
+
+    def calculate_percentage(self, numerator, denominator):
+    """Calculate percentage with safety check"""
+        try:
+            if denominator and denominator != 0:
+                 return (numerator / denominator) * 100
+            return 0.0
+        except:
+            return 0.0
+
+    def get_default_metrics(self):
+    """Return default metrics structure"""
+        return {
+        "Revenue": {"2025": 0, "2021": 0, "Growth": 0},
+        "Profit": {"2025": 0, "2021": 0, "Growth": 0, "Margin_2025": 0},
+        "Assets": {"2025": 0, "2021": 0, "Growth": 0}
+    }
+
+    def generate_analysis_prompt(self, metrics):
+    """Create focused analysis prompt with safety checks"""
+        return f"""<human>Analyze these financial metrics and provide insights:
+
+Key Performance Indicators:
+1. Revenue Performance:
+   - 2025: ${metrics['Revenue']['2025']:,.1f}M
+   - 2021: ${metrics['Revenue']['2021']:,.1f}M
+   - 5-Year Growth: {metrics['Revenue']['Growth']:.1f}%
+
+2. Profitability:
+   - 2025 Net Profit: ${metrics['Profit']['2025']:,.1f}M
+   - 2021 Net Profit: ${metrics['Profit']['2021']:,.1f}M
+   - Profit Growth: {metrics['Profit']['Growth']:.1f}%
+   - 2025 Profit Margin: {metrics['Profit']['Margin_2025']:.1f}%
+
+3. Asset Base:
+   - 2025 Total Assets: ${metrics['Assets']['2025']:,.1f}M
+   - 2021 Total Assets: ${metrics['Assets']['2021']:,.1f}M
+   - Asset Growth: {metrics['Assets']['Growth']:.1f}%
+
+Based on these metrics, provide:
+1. Financial Performance Assessment
+2. Key Strengths and Weaknesses
+3. Strategic Recommendations</human>"""
 
     def generate_analysis_prompt(self, metrics):
         """Create focused analysis prompt"""