Update app.py

app.py

@@ -7,9 +7,20 @@ import datetime
 import tempfile
 from transformers import pipeline
 
-# Initialize Summarizer and Chat Model
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-chat_model = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", max_length=256)
+
+# Lightweight model for quick answers
+light_chat_model = pipeline("text-generation", model="google/flan-t5-small", max_length=256)
+
+# Heavier model for advanced summarization
+advanced_summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+
+def answer_investing_question(question, level="basic"):
+    if level == "basic":
+        response = light_chat_model(question)[0]['generated_text']
+    else:
+        prompt = f"Explain in detail: {question}"
+        response = advanced_summarizer(prompt, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
+    return response.strip()
 
 # API Key
 POLYGON_API_KEY = os.getenv("POLYGON_API_KEY")
@@ -195,9 +206,11 @@ with gr.Blocks(theme="soft") as iface:
             output_chart = gr.Plot()
         with gr.Tab("Ask About Investing"):
             user_question = gr.Textbox(label="Ask about investing...")
-            answer_box = gr.Textbox()
+            level = gr.Dropdown(choices=["basic", "advanced"], value="basic", label="Answer Detail Level")
+            answer_box = gr.Textbox(label="Answer")
             ask_button = gr.Button("Get Answer")
-            ask_button.click(fn=answer_investing_question, inputs=[user_question], outputs=[answer_box])
+            ask_button.click(fn=answer_investing_question, inputs=[user_question, level], outputs=[answer_box])
+
 
     submit_btn = gr.Button("Run Analysis")
     download_btn = gr.Button("Download Report")