Update app.py

app.py

@@ -34,6 +34,16 @@ class FinancialAnalyzer:
         except:
             return 0.0
 
+    def is_valid_markdown(self, file_path):
+        """Check if a file is a valid Markdown file"""
+        try:
+            with open(file_path, 'r') as f:
+                content = f.read()
+            # Simple check for Markdown structure
+            return any(line.startswith('#') or '|' in line for line in content.split('\n'))
+        except:
+            return False
+
     def parse_financial_data(self, content):
         """Parse markdown content into structured data"""
         try:
@@ -122,7 +132,8 @@ class FinancialAnalyzer:
     def generate_analysis_prompt(self, metrics):
         """Create analysis prompt from metrics"""
         try:
-            return f"""<human>Analyze these financial metrics for 2025:
+            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
@@ -136,10 +147,8 @@ Balance Sheet Strength:
 - Total Liabilities: ${metrics['Balance_Sheet']['Total_Liabilities_2025']:,.1f}M
 - Shareholders' Equity: ${metrics['Balance_Sheet']['Equity_2025']:,.1f}M
 
-Provide a concise analysis of:
-1. Financial Health
-2. Key Strengths and Weaknesses
-3. Strategic Recommendations</human>"""
+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 ""
@@ -169,6 +178,10 @@ Provide a concise analysis of:
     def analyze_financials(self, balance_sheet_file, income_stmt_file):
         """Main analysis function"""
         try:
+            # Validate files
+            if not (self.is_valid_markdown(balance_sheet_file) and self.is_valid_markdown(income_stmt_file)):
+                return "Error: One or both files are invalid or not in Markdown format."
+
             # Read files
             with open(balance_sheet_file, 'r') as f:
                 balance_sheet = f.read()
@@ -219,4 +232,4 @@ def create_interface():
 
 if __name__ == "__main__":
     iface = create_interface()
-    iface.launch()
+    iface.launch()