Update app.py

app.py

@@ -163,6 +163,7 @@ def generate_summary(info, ratios):
         recommendation = "Buy"
     elif ratios['P/E Ratio'] > 30 and ratios['P/B Ratio'] > 5 and ratios['PEG Ratio'] > 2.0:
         recommendation = "Sell"
+    
     report = (
         f"Company Overview:\n"
         f"Name: {info['Name']}\n"
@@ -177,7 +178,10 @@ def generate_summary(info, ratios):
         f"Dividend Yield: {ratios['Dividend Yield']:.2f}%\n\n"
         f"Recommended Investment Action: {recommendation}.\n"
     )
-    return summarizer(report, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
+
+    # Use Mistral to generate the summary
+    summary_prompt = f"Summarize the following financial report clearly and briefly:\n\n{report}"
+    return query_mistral(summary_prompt)
 
 def stock_research(symbol, eps=5.0, growth=0.1, book=500000000):
     info = get_company_info(symbol)