Update app.py

app.py

@@ -7,7 +7,6 @@ import datetime
 import tempfile
 from transformers import pipeline
 
-
 # Your Hugging Face API Token (set this safely)
 HF_TOKEN = os.getenv("HUGGINGFACE_TOKEN")  # Recommended: use environment variable
 
@@ -19,6 +18,9 @@ headers = {
     "Authorization": f"Bearer {HF_TOKEN}"
 }
 
+# Load summarizer model
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+
 # Function to query Mistral API
 def query_mistral(question):
     payload = {
@@ -29,7 +31,6 @@ def query_mistral(question):
     output = response.json()
     return output[0]["generated_text"]
 
-
 # API Key
 POLYGON_API_KEY = os.getenv("POLYGON_API_KEY")
 
@@ -164,7 +165,6 @@ def generate_summary(info, ratios):
     )
     return summarizer(report, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
 
-
 def stock_research(symbol, eps=5.0, growth=0.1, book=500000000):
     info = get_company_info(symbol)
     price = get_current_price(symbol)
@@ -209,7 +209,7 @@ with gr.Blocks(theme="soft") as iface:
             output_sector = gr.Dataframe()
         with gr.Tab("Historical Price Chart"):
             output_chart = gr.Plot()
-        with gr.Tab("Ask About Investing"):   # <-- ✅ Correct (no extra Blocks)
+        with gr.Tab("Ask About Investing"):
             user_question = gr.Textbox(label="Ask about investing...")
             answer_box = gr.Textbox(label="Answer")
             ask_button = gr.Button("Get Answer")
@@ -222,6 +222,5 @@ with gr.Blocks(theme="soft") as iface:
     submit_btn.click(fn=stock_research, inputs=[symbol, eps, growth, book],
                      outputs=[output_summary, output_info, output_ratios, output_sector, output_chart])
 
-
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()