Update app.py

app.py

@@ -2,27 +2,13 @@ import os
 import requests
 import gradio as gr
 import openai
-import matplotlib.pyplot as plt
-from io import BytesIO
 
 # 🔐 Load secrets from environment
 client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 STOCK_API_KEY = os.getenv("STOCK_API_KEY")
 
-# 🔍 Improved stock symbol fetch with alias support
+# 🔍 Get stock symbol from company name
 def get_stock_symbol(company_name):
-    aliases = {
-        "facebook": "META",
-        "google": "GOOGL",
-        "tesla": "TSLA",
-        "apple": "AAPL",
-        "amazon": "AMZN"
-    }
-
-    name_lower = company_name.lower()
-    if name_lower in aliases:
-        return aliases[name_lower]
-
     url = 'https://www.alphavantage.co/query'
     params = {
         'function': 'SYMBOL_SEARCH',
@@ -31,12 +17,8 @@ def get_stock_symbol(company_name):
     }
     r = requests.get(url, params=params).json()
     try:
-        best_match = next(
-            match for match in r['bestMatches']
-            if company_name.lower() in match['2. name'].lower()
-        )
-        return best_match['1. symbol']
-    except (StopIteration, KeyError, IndexError):
+        return r['bestMatches'][0]['1. symbol']
+    except (KeyError, IndexError):
         return None
 
 # 📈 Get real-time stock price and last trading date
@@ -96,51 +78,21 @@ def ask_gpt_summary(company_name, financial_data):
     except Exception as e:
         return f"GPT Error: {e}"
 
-# 📊 Generate bar chart of financials
-def safe_float(value):
-    try:
-        return float(value)
-    except (ValueError, TypeError):
-        return 0
-
-def plot_financials(financial_data):
-    labels = ['P/E Ratio', 'EPS', 'Market Cap ($B)']
-    values = [
-        safe_float(financial_data['P/E Ratio']),
-        safe_float(financial_data['EPS']),
-        safe_float(financial_data['Market Cap']) / 1e9
-    ]
-    plt.figure(figsize=(6, 3))
-    bars = plt.barh(labels, values)
-    for label, bar in zip(labels, bars):
-        val = bar.get_width()
-        text = f"${val:.2f}B" if 'Market Cap' in label else f"{val:.2f}"
-        plt.text(val, bar.get_y() + bar.get_height() / 2, text, va='center')
-    plt.xlabel('Value')
-    plt.title('Company Financial Overview')
-    plt.tight_layout()
-    buf = BytesIO()
-    plt.savefig(buf, format='png')
-    plt.close()
-    buf.seek(0)
-    return buf
-
 # 🎯 Main function
 def stock_insight(company_name):
     symbol = get_stock_symbol(company_name)
     if not symbol:
-        return f"❌ Could not find stock symbol for '{company_name}'.", "", "", "", "", "", None
+        return f"❌ Could not find stock symbol for '{company_name}'.", "", "", "", "", ""
 
     price, last_trade = get_stock_quote(symbol)
     if not price:
-        return f"❌ Could not fetch price for symbol {symbol}.", "", "", "", "", "", None
+        return f"❌ Could not fetch price for symbol {symbol}.", "", "", "", "", ""
 
     overview = get_financial_overview(symbol)
     if not overview:
-        return f"❌ Could not fetch financial data for {symbol}.", "", "", "", "", "", None
+        return f"❌ Could not fetch financial data for {symbol}.", "", "", "", "", ""
 
     summary = ask_gpt_summary(company_name, overview)
-    chart_buffer = plot_financials(overview)
 
     return (
         f"✅ Symbol: {symbol}",
@@ -148,32 +100,23 @@ def stock_insight(company_name):
         f"🗓️ Last Trading Day: {last_trade}",
         f"📊 P/E Ratio: {overview['P/E Ratio']}\nEPS: {overview['EPS']}\nMarket Cap: ${overview['Market Cap']}",
         f"📝 Company Summary: {overview['Description'][:300]}...",
-        f"🤖 GPT Insight: {summary}",
-        chart_buffer
+        f"🤖 GPT Insight: {summary}"
     )
 
-# 🖥️ Gradio Interface with Blocks
-with gr.Blocks() as demo:
-    gr.Markdown("# 📈 Stock Market Assistant")
-    with gr.Row():
-        company_input = gr.Textbox(label="Enter Company Name")
-        search_button = gr.Button("🔍 Search")
-
-    with gr.Row():
-        with gr.Column():
-            symbol_output = gr.Textbox(label="Symbol")
-            price_output = gr.Textbox(label="Stock Price")
-            trade_output = gr.Textbox(label="Last Trading Day")
-            financial_output = gr.Textbox(label="Financial Overview")
-        with gr.Column():
-            chart_output = gr.Image(label="📊 Financial Chart", type="pil")
-
-    company_summary = gr.Textbox(label="Company Summary")
-    gpt_output = gr.Textbox(label="AI Insight")
-
-    search_button.click(stock_insight, inputs=company_input, outputs=[
-        symbol_output, price_output, trade_output,
-        financial_output, company_summary, gpt_output, chart_output
-    ])
-
-    demo.launch()
+# 🖥️ Gradio Interface
+demo = gr.Interface(
+    fn=stock_insight,
+    inputs=gr.Textbox(label="Enter Company Name"),
+    outputs=[
+        gr.Textbox(label="Symbol"),
+        gr.Textbox(label="Stock Price"),
+        gr.Textbox(label="Last Trading Day"),
+        gr.Textbox(label="Financial Overview"),
+        gr.Textbox(label="Company Summary"),
+        gr.Textbox(label="AI Insight")
+    ],
+    title="📈 Stock Market Assistant",
+    description="Search by company name to get stock symbol, live price, financial info, last trade date, and AI-based insight."
+)
+
+demo.launch()