Update app.py

app.py

@@ -1,3 +1,84 @@
+# Add these imports at top
+import yfinance as yf
+from datetime import datetime, timedelta
+from transformers import pipeline
+import torch
+
+# Add this function
+def get_ai_recommendations(portfolio_stocks):
+    """Generate AI recommendations for portfolio"""
+    recommendations = []
+    
+    for stock in portfolio_stocks:
+        try:
+            # Get stock data
+            ticker = yf.Ticker(stock['symbol'])
+            hist = ticker.history(period="6mo")
+            
+            if len(hist) > 30:
+                # Technical indicators
+                current_price = hist['Close'].iloc[-1]
+                sma_30 = hist['Close'].tail(30).mean()
+                sma_200 = hist['Close'].tail(200).mean() if len(hist) > 200 else sma_30
+                
+                # AI recommendation logic
+                if current_price > sma_200 * 1.1:
+                    rec = "STRONG BUY"
+                    reason = "Well above 200D MA, bullish trend"
+                elif current_price < sma_200 * 0.9:
+                    rec = "SELL"
+                    reason = "Below 200D MA, bearish trend"
+                elif current_price > sma_30:
+                    rec = "BUY"
+                    reason = "Above 30D MA, positive momentum"
+                else:
+                    rec = "HOLD"
+                    reason = "Neutral position, wait for breakout"
+                
+                recommendations.append({
+                    'symbol': stock['symbol'],
+                    'recommendation': rec,
+                    'reason': reason,
+                    'current_price': current_price,
+                    'sma_30': sma_30,
+                    'sma_200': sma_200
+                })
+                
+        except Exception as e:
+            print(f"Error analyzing {stock['symbol']}: {e}")
+    
+    return recommendations
+
+# In your main app, add AI section
+st.sidebar.title("🤖 AI Stock Analyst")
+
+if st.sidebar.button("Run AI Analysis on Portfolio"):
+    with st.spinner("AI analyzing your portfolio..."):
+        # Load your portfolio
+        portfolio = [...]  # Your 39 stocks
+        
+        # Get AI recommendations
+        ai_recs = get_ai_recommendations(portfolio)
+        
+        # Display results
+        st.subheader("🧠 AI Portfolio Analysis")
+        
+        for rec in ai_recs[:10]:  # Show top 10
+            col1, col2, col3 = st.columns([1, 2, 1])
+            with col1:
+                st.write(f"**{rec['symbol']}**")
+            with col2:
+                if rec['recommendation'] == 'STRONG BUY':
+                    st.success(f"🎯 {rec['recommendation']}")
+                elif rec['recommendation'] == 'SELL':
+                    st.error(f"⚠️ {rec['recommendation']}")
+                else:
+                    st.info(f"📊 {rec['recommendation']}")
+                st.caption(rec['reason'])
+            with col3:
+                st.metric("Price", f"${rec['current_price']:.2f}")
+
+
 # Add at top
 from transformers import pipeline