Update app.py

app.py

@@ -1,42 +1,17 @@
-import yfinance as yf
 import gradio as gr
-from langchain.chains import LLMChain
-from langchain.prompts import PromptTemplate
-from langchain.llms import LlamaCpp
-
-# Define Llama model (you must provide the path to your Llama model)
-llama_model = LlamaCpp(model_path="path_to_your_llama_model.bin")
-
-# Set up Langchain LLM
-llm_chain = LLMChain(llm=llama_model)
-
-# Define the prompt template for querying the LLM about stock details
-prompt_template = """
-You are a stock market assistant. Please provide detailed information about the stock.
-The stock symbol is: {stock_symbol}.
-"""
-stock_prompt = PromptTemplate(input_variables=["stock_symbol"], template=prompt_template)
-
-# Function to get stock data from Yahoo Finance
-def get_stock_data(stock_symbol):
-    stock = yf.Ticker(stock_symbol)
-    stock_info = stock.info  # Get company info
-    stock_price = stock_info['currentPrice']  # Get current stock price
-    company_name = stock_info.get('shortName', 'N/A')
-    sector = stock_info.get('sector', 'N/A')
-    market_cap = stock_info.get('marketCap', 'N/A')
-
-    # Generate a response using Langchain LLM
-    stock_details = llm_chain.run(stock_prompt.format(stock_symbol=stock_symbol))
-    
-    return f"Company: {company_name}\nSector: {sector}\nMarket Cap: {market_cap}\nStock Price: ${stock_price}\n\nDetailed Information: {stock_details}"
-
-# Gradio Interface
-def stock_market_agent(stock_symbol):
-    return get_stock_data(stock_symbol)
-
-# Set up Gradio UI
-iface = gr.Interface(fn=stock_market_agent,
+from yahoo_fin import stock_info
+
+def get_stock_info(symbol):
+    try:
+        # Fetch real-time stock price
+        price = stock_info.get_live_price(symbol)
+        # Fetch company information
+        company_info = stock_info.get_quote_table(symbol)
+        return f"Company: {company_info['Name']}\nSector: {company_info['Sector']}\nMarket Cap: {company_info['Market Cap']}\nStock Price: ${price:.2f}"
+    except Exception as e:
+        return f"Error retrieving data: {e}"
+
+iface = gr.Interface(fn=get_stock_info,
                      inputs=gr.Textbox(label="Enter Stock Symbol (e.g., AAPL)"),
                      outputs="text",
                      title="Stock Market Agent",