added relevent documents

.gitignore

@@ -0,0 +1 @@
+.env

app.py

@@ -1,4 +1,85 @@
 import streamlit as st
+import yfinance as yf
+import requests
+import pandas as pd
+from langchain.agents import initialize_agent, AgentType
+from langchain.tools import Tool
+from langchain_huggingface import HuggingFacePipeline
+import os
+from dotenv import load_dotenv
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+load_dotenv()
+NEWSAPI_KEY = os.getenv("NEWSAPI_KEY")
+
+def fetch_stock_data(ticker):
+    stock = yf.Ticker(ticker)
+    hist = stock.history(period="1mo")
+    return hist.tail(5)
+
+def fetch_stock_news(ticker, NEWSAPI_KEY):
+    api_url = f"https://newsapi.org/v2/everything?q={ticker}&apiKey={NEWSAPI_KEY}"
+    response = requests.get(api_url)
+    if response.status_code == 200:
+        articles = response.json().get('articles', [])
+        return [{"title": article['title'], "description": article['description']} for article in articles[:5]]
+    else:
+        return [{"error": "Unable to fetch news."}]
+
+def calculate_moving_average(ticker, window=5):
+    stock = yf.Ticker(ticker)
+    hist = stock.history(period="1mo")
+    hist[f"{window}-day MA"] = hist["Close"].rolling(window=window).mean()
+    return hist[["Close", f"{window}-day MA"]].tail(5)
+
+llm = HuggingFacePipeline.from_model_id(
+    model_id="microsoft/Phi-3-mini-4k-instruct",
+    task="text-generation",
+    pipeline_kwargs={
+        "max_new_tokens": 100,
+        "top_k": 50,
+        "temperature": 0.1,
+    },
+)
+
+stock_data_tool = Tool(
+    name="Stock Data Fetcher",
+    func=fetch_stock_data,
+    description="Fetch recent stock data for a given ticker."
+)
+
+stock_news_tool = Tool(
+    name="Stock News Fetcher",
+    func=lambda ticker: fetch_stock_news(ticker, NEWSAPI_KEY),
+    description="Fetch recent news articles about a stock ticker."
+)
+
+moving_average_tool = Tool(
+    name="Moving Average Calculator",
+    func=calculate_moving_average,
+    description="Calculate the moving average of a stock over a 5-day window."
+)
+
+tools = [stock_data_tool, stock_news_tool, moving_average_tool]
+
+agent = initialize_agent(
+    tools=tools,
+    llm=llm,
+    agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+    verbose=True
+)
+
+st.title("Trading Helper Agent")
+
+query = st.text_input("Enter your query:")
+
+if st.button("Submit"):
+    if query:
+        with st.spinner("Processing..."):
+            try:
+                response = agent.run(query)
+                st.success("Response:")
+                st.write(response)
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+    else:
+        st.warning("Please enter a query.")

requirements.txt

@@ -0,0 +1,7 @@
+streamlit
+yfinance
+requests
+pandas
+langchain
+langchain_huggingface
+python-dotenv