Update app.py

app.py

@@ -18,9 +18,6 @@ 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 = {
@@ -42,6 +39,23 @@ sector_averages = {
     "Energy": {"P/E Ratio": 12, "P/S Ratio": 1.2, "P/B Ratio": 1.3},
 }
 
+# Safe Request Function
+def safe_request(url):
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+        return response
+    except:
+        return None
+
+# Fetch Functions
+# (functions unchanged)
+# ...
+
+# Financial Calculations
+# (functions unchanged)
+# ...
+
 # Safe Request Function
 def safe_request(url):
     try:
@@ -191,7 +205,9 @@ def stock_research(symbol, eps=5.0, growth=0.1, book=500000000):
     return summary, info_table, ratios_table, sector_comp, fig
 
 # Gradio UI
-with gr.Blocks(theme="soft") as iface:
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as iface:
+    theme_choice = gr.Radio(choices=["light", "dark"], value="light", label="Choose Theme")
+
     with gr.Row():
         symbol = gr.Textbox(label="Stock Symbol (e.g., AAPL)")
         eps = gr.Number(label="Assumed EPS", value=5.0)
@@ -213,7 +229,15 @@ with gr.Blocks(theme="soft") as iface:
             user_question = gr.Textbox(label="Ask about investing...")
             answer_box = gr.Textbox(label="Answer")
             ask_button = gr.Button("Get Answer")
-            ask_button.click(fn=query_mistral, inputs=[user_question], outputs=[answer_box])
+            with gr.Row():
+                ask_button.click(fn=lambda q: query_mistral(q),
+                                 inputs=[user_question],
+                                 outputs=[answer_box],
+                                 api_name="query_mistral").then(
+                    lambda: "",
+                    inputs=[],
+                    outputs=[user_question]
+                )
 
     submit_btn = gr.Button("Run Analysis")
     download_btn = gr.Button("Download Report")
@@ -222,5 +246,6 @@ 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()