Update app.py

app.py

@@ -92,66 +92,67 @@ class FinancialAnalyzer:
             return {}
 
     def extract_metrics(self, income_data, balance_data):
-        """Extract and calculate key financial metrics"""
-        try:
-            metrics = {
-                "Revenue": {
-                    "2025": self.get_nested_value(income_data, "Revenue", "Total Net Revenue", "2025"),
-                    "2021": self.get_nested_value(income_data, "Revenue", "Total Net Revenue", "2021")
-                },
-                "Profitability": {
-                    "Gross_Profit_2025": self.get_nested_value(income_data, "Cost and Gross Profit", "Gross Profit", "2025"),
-                    "Net_Earnings_2025": self.get_nested_value(income_data, "Profit Summary", "Net Earnings", "2025"),
-                    "Operating_Expenses_2025": self.get_nested_value(income_data, "Operating Expenses", "Total Operating Expenses", "2025")
-                },
-                "Balance_Sheet": {
-                    "Total_Assets_2025": self.get_nested_value(balance_data, "Key Totals", "Total_Assets", "2025"),
-                    "Total_Liabilities_2025": self.get_nested_value(balance_data, "Key Totals", "Total_Liabilities", "2025"),
-                    "Equity_2025": self.get_nested_value(balance_data, "Key Totals", "Total_Shareholders_Equity", "2025")
-                }
+    """Extract and calculate key financial metrics"""
+    try:
+        # Get current and previous year values for growth calculations
+        metrics = {
+            "Revenue": {
+                "2025": self.get_nested_value(income_data, "Revenue", "Total Net Revenue", "2025"),
+                "2024": self.get_nested_value(income_data, "Revenue", "Total Net Revenue", "2024"),
+                "2021": self.get_nested_value(income_data, "Revenue", "Total Net Revenue", "2021")
+            },
+            "Profitability": {
+                "Gross_Profit_2025": self.get_nested_value(income_data, "Cost and Gross Profit", "Gross Profit", "2025"),
+                "EBIT_2025": self.get_nested_value(income_data, "Profit Summary", "EBIT", "2025"),
+                "Net_Earnings_2025": self.get_nested_value(income_data, "Profit Summary", "Net Earnings", "2025"),
+                "Operating_Expenses_2025": self.get_nested_value(income_data, "Operating Expenses", "Total Operating Expenses", "2025")
+            },
+            "Balance_Sheet": {
+                "Total_Assets_2025": self.get_nested_value(balance_data, "Key Totals", "Total_Assets", "2025"),
+                "Current_Assets_2025": self.get_nested_value(balance_data, "Key Totals", "Total_Current_Assets", "2025"),
+                "Total_Liabilities_2025": self.get_nested_value(balance_data, "Key Totals", "Total_Liabilities", "2025"),
+                "Current_Liabilities_2025": self.get_nested_value(balance_data, "Key Totals", "Total_Current_Liabilities", "2025"),
+                "Equity_2025": self.get_nested_value(balance_data, "Key Totals", "Total_Shareholders_Equity", "2025"),
+                "Inventory_2025": self.get_nested_value(balance_data, "Balance Sheet Data 2021-2025", "Inventory", "2025"),
+                "Accounts_Receivable_2025": self.get_nested_value(balance_data, "Balance Sheet Data 2021-2025", "Accounts_Receivable", "2025"),
+                "Long_Term_Debt_2025": self.get_nested_value(balance_data, "Balance Sheet Data 2021-2025", "Long_Term_Debt", "2025")
+            },
+            "Cash_Flow": {
+                "Depreciation_2025": self.get_nested_value(income_data, "Operating Expenses", "Depreciation & Amortization", "2025"),
+                "Interest_Expense_2025": self.get_nested_value(income_data, "Profit Summary", "Interest Expense", "2025")
             }
-            
-            # Calculate additional metrics
-            revenue_2025 = metrics["Revenue"]["2025"]
-            if revenue_2025 != 0:
-                metrics["Profitability"]["Gross_Margin"] = (metrics["Profitability"]["Gross_Profit_2025"] / revenue_2025) * 100
-                metrics["Profitability"]["Net_Margin"] = (metrics["Profitability"]["Net_Earnings_2025"] / revenue_2025) * 100
-            
-            return metrics
-        except Exception as e:
-            print(f"Error extracting metrics: {str(e)}")
-            return {}
-
-    def get_nested_value(self, data, section, key, year):
-        """Safely get nested dictionary value"""
-        try:
-            return data.get(section, {}).get(key, {}).get(year, 0)
-        except:
-            return 0
-
-    def generate_analysis_prompt(self, metrics):
-        """Create analysis prompt from metrics"""
-        try:
-            return f"""<human>
-Analyze these financial metrics for 2025 with a focus on business performance, trends, and risks:
-
-Revenue and Profitability:
-- Total Revenue: ${metrics['Revenue']['2025']:,.1f}M
-- Gross Profit: ${metrics['Profitability']['Gross_Profit_2025']:,.1f}M
-- Net Earnings: ${metrics['Profitability']['Net_Earnings_2025']:,.1f}M
-- Gross Margin: {metrics['Profitability'].get('Gross_Margin', 0):,.1f}%
-- Net Margin: {metrics['Profitability'].get('Net_Margin', 0):,.1f}%
-
-Balance Sheet Strength:
-- Total Assets: ${metrics['Balance_Sheet']['Total_Assets_2025']:,.1f}M
-- Total Liabilities: ${metrics['Balance_Sheet']['Total_Liabilities_2025']:,.1f}M
-- Shareholders' Equity: ${metrics['Balance_Sheet']['Equity_2025']:,.1f}M
-
-Explain key financial ratios and their implications. Discuss strategies for growth and risk mitigation.
-</human>"""
-        except Exception as e:
-            print(f"Error generating prompt: {str(e)}")
-            return ""
+        }
+        
+        # Calculate financial ratios
+        revenue_2025 = metrics["Revenue"]["2025"]
+        if revenue_2025 != 0:
+            # Profitability Ratios
+            metrics["Ratios"] = {
+                "Gross_Margin": (metrics["Profitability"]["Gross_Profit_2025"] / revenue_2025) * 100,
+                "Operating_Margin": (metrics["Profitability"]["EBIT_2025"] / revenue_2025) * 100,
+                "Net_Margin": (metrics["Profitability"]["Net_Earnings_2025"] / revenue_2025) * 100,
+                
+                # Liquidity Ratios
+                "Current_Ratio": metrics["Balance_Sheet"]["Current_Assets_2025"] / metrics["Balance_Sheet"]["Current_Liabilities_2025"],
+                "Quick_Ratio": (metrics["Balance_Sheet"]["Current_Assets_2025"] - metrics["Balance_Sheet"]["Inventory_2025"]) / metrics["Balance_Sheet"]["Current_Liabilities_2025"],
+                
+                # Efficiency Ratios
+                "Asset_Turnover": revenue_2025 / metrics["Balance_Sheet"]["Total_Assets_2025"],
+                "Receivables_Turnover": revenue_2025 / metrics["Balance_Sheet"]["Accounts_Receivable_2025"],
+                
+                # Solvency Ratios
+                "Debt_to_Equity": metrics["Balance_Sheet"]["Total_Liabilities_2025"] / metrics["Balance_Sheet"]["Equity_2025"],
+                "Interest_Coverage": metrics["Profitability"]["EBIT_2025"] / metrics["Cash_Flow"]["Interest_Expense_2025"] if metrics["Cash_Flow"]["Interest_Expense_2025"] != 0 else 0,
+                
+                # Growth Rates
+                "Revenue_Growth": ((metrics["Revenue"]["2025"] / metrics["Revenue"]["2024"]) - 1) * 100,
+                "5Year_Revenue_CAGR": ((metrics["Revenue"]["2025"] / metrics["Revenue"]["2021"]) ** (1/4) - 1) * 100
+            }
+        
+        return metrics
+    except Exception as e:
+        print(f"Error extracting metrics: {str(e)}")
+        return {}
 
     def generate_analysis(self, prompt):
         """Generate analysis using TinyLlama"""
@@ -160,7 +161,7 @@ Explain key financial ratios and their implications. Discuss strategies for grow
             
             outputs = self.model.generate(
                 inputs["input_ids"],
-                max_new_tokens=500,  # Generate up to 500 new tokens
+                max_new_tokens=800,  # Increased from 500 to 800 for more detailed analysis
                 temperature=0.7,
                 top_p=0.9,
                 do_sample=True,