Update app.py

app.py

@@ -1,114 +1,145 @@
 import gradio as gr
 import yfinance as yf
 import pandas as pd
+import numpy as np
 import plotly.graph_objects as go
 from transformers import pipeline
-from datetime import datetime, timedelta
+from datetime import datetime
 import requests
 from bs4 import BeautifulSoup
 import feedparser
 
-# ------------------- Hugging Face Models -------------------
+# ------------------- Initialize Models -------------------
 sentiment_analyzer = pipeline("text-classification", model="ProsusAI/finbert")
 news_summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-ner_model = pipeline("ner", model="dslim/bert-base-NER")
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ur-en")
 
-# ------------------- Technical Analysis -------------------
+# ------------------- Technical Analysis Functions -------------------
 def calculate_rsi(data, window=14):
     delta = data['Close'].diff()
     gain = (delta.where(delta > 0, 0)).rolling(window=window).mean()
     loss = (-delta.where(delta < 0, 0)).rolling(window=window).mean()
     rs = gain / loss
-    rsi = 100 - (100 / (1 + rs))
-    return rsi
+    return 100 - (100 / (1 + rs))
 
-# ------------------- Fetch PSX Data -------------------
-def get_stock_data(ticker, period="1y"):
+def calculate_macd(data, slow=26, fast=12, signal=9):
+    exp1 = data['Close'].ewm(span=fast, adjust=False).mean()
+    exp2 = data['Close'].ewm(span=slow, adjust=False).mean()
+    macd = exp1 - exp2
+    signal_line = macd.ewm(span=signal, adjust=False).mean()
+    return macd, signal_line
+
+# ------------------- Data Fetching & Processing -------------------
+def get_psx_data(ticker):
     try:
-        stock = yf.Ticker(f"{ticker}.KAR")
-        data = stock.history(period=period)
+        stock = yf.Ticker(f"{ticker}.KA")
+        data = stock.history(period="1y")
         if data.empty:
-            return None, "Error: Ticker not found or no data."
+            return None, "Invalid ticker or no data available"
+            
+        # Calculate indicators
         data['RSI'] = calculate_rsi(data)
+        data['MACD'], data['Signal'] = calculate_macd(data)
+        data['MA50'] = data['Close'].rolling(window=50).mean()
+        data['MA200'] = data['Close'].rolling(window=200).mean()
+        
         return data, None
     except Exception as e:
-        return None, f"Error: {str(e)}"
-
-# ------------------- Sentiment & News Analysis -------------------
-def analyze_sentiment(text):
-    return sentiment_analyzer(text)[0]['label']
-
-def summarize_news(url):
-    response = requests.get(url)
-    soup = BeautifulSoup(response.text, 'html.parser')
-    paragraphs = soup.find_all('p')
-    article = ' '.join([p.get_text() for p in paragraphs[:5]])
-    summary = news_summarizer(article, max_length=100, min_length=30)[0]['summary_text']
-    return summary
+        return None, str(e)
 
-def get_geo_news():
-    feed = feedparser.parse("https://www.dawn.com/feeds/pakistan")
-    news = []
-    for entry in feed.entries[:3]:
-        summary = summarize_news(entry.link)
-        sentiment = analyze_sentiment(entry.title)
-        news.append({"title": entry.title, "summary": summary, "sentiment": sentiment})
-    return news
-
-# ------------------- Generate Signal -------------------
-def generate_signal(rsi, sentiment_score, news_score):
-    if rsi < 30 and sentiment_score > 0.5 and news_score > 0.5:
-        return "STRONG BUY 🟢"
-    elif rsi > 70 or sentiment_score < -0.5:
-        return "STRONG SELL 🔴"
-    else:
-        return "HOLD 🟡"
+# ------------------- News & Sentiment Analysis -------------------
+def analyze_psx_news():
+    news_items = []
+    try:
+        feed = feedparser.parse("https://www.dawn.com/feeds/pakistan")
+        for entry in feed.entries[:5]:  # Limit to 5 articles
+            try:
+                # Translate Urdu content to English
+                if any(char in entry.title for char in ['\u0600', '\u0800']):  # Detect Urdu characters
+                    translated = translator(entry.title)[0]['translation_text']
+                    title = f"[URDU] {translated}"
+                else:
+                    title = entry.title
+                
+                # Analyze sentiment
+                sentiment = sentiment_analyzer(title)[0]
+                news_items.append({
+                    'title': title,
+                    'sentiment': sentiment['label'],
+                    'score': sentiment['score'],
+                    'link': entry.link
+                })
+            except:
+                continue
+    except:
+        pass
+    return news_items
 
 # ------------------- Gradio Interface -------------------
 def analyze_stock(ticker):
-    # Fetch Data
-    data, error = get_stock_data(ticker)
+    # Get stock data
+    data, error = get_psx_data(ticker)
     if error:
-        return error, None, None
-    
-    # Technical Analysis
-    latest_rsi = data['RSI'].iloc[-1]
-    
-    # Sentiment & News Analysis
-    news = get_geo_news()
-    sentiment_scores = [1 if n['sentiment'] == 'positive' else -1 for n in news]
-    sentiment_score = sum(sentiment_scores) / len(sentiment_scores) if sentiment_scores else 0
-    
-    # Generate Signal
-    signal = generate_signal(latest_rsi, sentiment_score, 0.5)
-    
-    # Plot
-    fig = go.Figure(data=[go.Candlestick(x=data.index,
-                open=data['Open'], high=data['High'],
-                low=data['Low'], close=data['Close'])])
-    fig.update_layout(title=f"{ticker} Price Chart")
-    
-    # News Summary
-    news_html = "<h3>Latest Geopolitical News:</h3>"
-    for n in news:
-        news_html += f"<p><b>{n['title']}</b> ({n['sentiment']})<br>{n['summary']}</p>"
-    
-    return signal, fig.to_html(), news_html
+        return error, None, None, None
+
+    # Create price chart
+    fig = go.Figure()
+    fig.add_trace(go.Candlestick(x=data.index,
+                open=data['Open'],
+                high=data['High'],
+                low=data['Low'],
+                close=data['Close'],
+                name='Price'))
+    fig.add_trace(go.Scatter(x=data.index, y=data['MA50'], line=dict(color='orange', width=1), name='MA50'))
+    fig.add_trace(go.Scatter(x=data.index, y=data['MA200'], line=dict(color='blue', width=1), name='MA200'))
+    fig.update_layout(title=f"{ticker} Technical Analysis", xaxis_rangeslider_visible=False)
+
+    # Generate signals
+    latest = data.iloc[-1]
+    rsi_signal = "Oversold" if latest['RSI'] < 30 else "Overbought" if latest['RSI'] > 70 else "Neutral"
+    macd_signal = "Bullish" if latest['MACD'] > latest['Signal'] else "Bearish"
+
+    # News analysis
+    news = analyze_psx_news()
+    news_html = "<h3>Market News & Sentiment</h3>"
+    for item in news:
+        news_html += f"""
+        <div style="margin-bottom: 15px; padding: 10px; border: 1px solid #eee; border-radius: 5px;">
+            <b>{item['title']}</b><br>
+            Sentiment: {item['sentiment']} ({item['score']:.2f})<br>
+            <a href="{item['link']}" target="_blank" style="color: #007bff;">Read more</a>
+        </div>
+        """
+
+    # Generate recommendation
+    recommendation = f"""
+    <div style="padding: 20px; background: {'#d4edda' if macd_signal == 'Bullish' else '#f8d7da'}; border-radius: 5px;">
+        <h4>Recommendation for {ticker}</h4>
+        <p>RSI: {latest['RSI']:.2f} ({rsi_signal})</p>
+        <p>MACD: {macd_signal}</p>
+        <p>50-Day MA: {latest['MA50']:.2f}</p>
+        <p>200-Day MA: {latest['MA200']:.2f}</p>
+        <h4 style="color: {'green' if macd_signal == 'Bullish' else 'red'};">{'Consider Buying' if macd_signal == 'Bullish' else 'Consider Selling'}</h4>
+    </div>
+    """
+
+    return recommendation, fig.to_html(), news_html
 
-# ------------------- Launch Gradio App -------------------
+# ------------------- App Configuration -------------------
 iface = gr.Interface(
     fn=analyze_stock,
-    inputs=gr.Textbox(label="Enter PSX Stock Ticker (e.g., HBL, LUCK)"),
+    inputs=gr.Textbox(label="Enter PSX Ticker (e.g., HBL, LUCK, OGDC)"),
     outputs=[
-        gr.Textbox(label="Recommendation"),
-        gr.HTML(label="Price Chart"),
-        gr.HTML(label="News Analysis")
+        gr.HTML(label="Recommendation"),
+        gr.HTML(label="Technical Analysis"),
+        gr.HTML(label="Market News")
     ],
-    title="🟢 PSX Stock Analysis Chatbot 🔴",
-    description="Real-time Technical, Sentiment, and News Analysis for Pakistan Stock Exchange",
-    allow_flagging="never"
+    title="🚀 PSX Smart Trading Assistant",
+    description="Real-time Pakistan Stock Exchange Analysis with AI-powered Insights",
+    examples=[["HBL"], ["LUCK"], ["OGDC"]],
+    allow_flagging="never",
+    theme=gr.themes.Soft()
 )
 
 if __name__ == "__main__":
-    iface.launch(share=True)  # Set share=True for public URL
+    iface.launch(server_name="0.0.0.0", server_port=7860)