chore: add improvements

Author: Shrividya

.github/workflows/python-app.yaml

@@ -23,6 +23,10 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt
+      - name: Run with black
+        run:
+          pip install black
+          black . --max-line-length=78
       - name: Lint with flake8
         run: |
           pip install flake8

src/__init__.py

@@ -27,7 +27,9 @@
     data = pd.read_csv("data/sample_stock.csv")
     data.columns = data.columns.str.strip()  # Remove any spaces
     print(data.columns)
-    data = pd.read_csv("data/sample_stock.csv", parse_dates=["Date"], index_col="Date")
+    data = pd.read_csv(
+        "data/sample_stock.csv", parse_dates=["Date"], index_col="Date"
+    )
 st.line_chart(data["Close"])
 st.subheader("Latest News Sentiment")
 sentiment_results = []

src/app.py

@@ -7,14 +7,23 @@
 
 
 def fetch_news(query, page_size=10):
-    """Fetch latest news articles using NewsAPI"""
     api_key = os.getenv("NEWS_API_KEY")
     if not api_key:
-        raise ValueError("Please set NEWS_API_KEY in .env file")
-    url = (
-        f"https://newsapi.org/v2/everything?q={query}&"
-        f"pageSize={page_size}&apiKey={api_key}"
-    )
-    response = requests.get(url)
+        raise RuntimeError("NEWS_API_KEY not found")
+
+    url = "https://newsapi.org/v2/everything"
+    params = {
+        "q": query,
+        "pageSize": page_size,
+        "apiKey": api_key,
+        "language": "en",
+        "sortBy": "publishedAt",
+    }
+
+    headers = {"User-Agent": "StockSentimentAnalyzer/1.0"}
+
+    response = requests.get(url, params=params, headers=headers)
+    response.raise_for_status()
+
     articles = response.json().get("articles", [])
-    return [(a["title"], a["description"]) for a in articles]
+    return [(a["title"], a.get("description", "")) for a in articles]

src/fetch_news.py

@@ -1,13 +1,50 @@
 import matplotlib.pyplot as plt
+import matplotlib.dates as mdates
 import streamlit as st
 
 
 def plot_stock(data, ticker):
     plt.figure(figsize=(10, 5))
-    plt.plot(data.index, data["Close"], label=f"{ticker} Close Price")
-    plt.xlabel("Time")
+
+    # Get today's high, low, and current price
+    today_high = data["High"].iloc[-1]
+    today_low = data["Low"].iloc[-1]
+    current_price = data["Close"].iloc[-1]
+
+    # Plot current price as a horizontal line
+    plt.hlines(
+        current_price,
+        xmin=data.index[0],
+        xmax=data.index[-1],
+        colors="red",
+        linewidth=2,
+        label=f"Current Price: {current_price:.2f}",
+    )
+
+    # Set y-axis limits to today's low/high with padding
+    plt.ylim(today_low * 0.995, today_high * 1.005)
+
+    # Format x-axis with ticks every 2 hours
+    ax = plt.gca()
+
+    # 2-hour interval
+    ax.xaxis.set_major_locator(mdates.HourLocator(interval=2))
+    ax.xaxis.set_major_formatter(mdates.DateFormatter("%H:%M"))
+    plt.xticks(rotation=45)
+
+    # Annotate current price
+    plt.text(
+        data.index[len(data) // 2],
+        current_price * 1.001,
+        f"{current_price:.2f}",
+        color="red",
+        fontsize=12,
+        ha="center",
+    )
+
     plt.ylabel("Price")
-    plt.title(f"{ticker} Stock Price")
+    plt.xlabel("Time (Market Hours)")
+    plt.title(f"{ticker} Price Today")
     plt.legend()
     st.pyplot(plt)