Update app.py

app.py

@@ -20,6 +20,66 @@ import re
 import json
 import os
 from googleapiclient.discovery import build
+import firebase_admin
+from firebase_admin import credentials, firestore
+from datetime import datetime
+
+# Firebase config
+firebase_config = {
+    "apiKey": "AIzaSyAU02LKQngy8e7JUAzwH7DY8HY6fdWYtPI",
+    "authDomain": "stock-predictor-566ce.firebaseapp.com",
+    "databaseURL": "https://stock-predictor-566ce-default-rtdb.asia-southeast1.firebasedatabase.app",
+    "projectId": "stock-predictor-566ce",
+    "storageBucket": "stock-predictor-566ce.appspot.com",
+    "messagingSenderId": "YOUR_SENDER_ID",
+    "appId": "YOUR_APP_ID"
+}
+
+firebase = pyrebase.initialize_app(firebase_config)
+auth = firebase.auth()
+
+cred = credentials.Certificate("firebase-service-key.json")
+firebase_admin.initialize_app(cred)
+firestore_db = firestore.client()
+
+current_uid = None  # To track logged in user
+
+# --- Part 2: Auth Functions ---
+def signup_user(email, password):
+    try:
+        user = auth.create_user_with_email_and_password(email, password)
+        auth.send_email_verification(user['idToken'])
+        return "✅ Signup successful! Please verify your email."
+    except Exception as e:
+        return f"❌ Signup error: {str(e)}"
+
+def login_user(email, password):
+    global current_uid
+    try:
+        user = auth.sign_in_with_email_and_password(email, password)
+        user_info = auth.get_account_info(user['idToken'])
+        current_uid = user_info['users'][0]['localId']
+        return current_uid, "✅ Login successful!"
+    except Exception as e:
+        return None, f"❌ Login error: {str(e)}"
+
+def logout_user():
+    global current_uid
+    current_uid = None
+    return "👋 Logged out."
+
+# --- Part 3: History Functions ---
+def save_user_history(uid, action_type, data):
+    firestore_db.collection("user_history").add({
+        "uid": uid,
+        "type": action_type,  # 'prediction' or 'portfolio'
+        "data": data,
+        "timestamp": datetime.now().isoformat()
+    })
+
+def get_user_history(uid):
+    docs = firestore_db.collection("user_history").where("uid", "==", uid).stream()
+    return [doc.to_dict() for doc in docs]
 
 # Load the LSTM model
 model = tf.keras.models.load_model("nse_lstm_model_fixed.h5")
@@ -689,6 +749,18 @@ Diversification, review cycles (quarterly), and informed decisions based on pred
 with gr.Blocks(css=custom_css) as app:
     with gr.Tabs():
 
+        # 🔐 Login Tab
+        with gr.Tab("🔐 Login / Signup"):
+            gr.Markdown("### 🔐 Welcome to Trade Sense")
+            with gr.Row():
+                email_input = gr.Textbox(label="Email")
+                password_input = gr.Textbox(label="Password", type="password")
+            with gr.Row():
+                login_btn = gr.Button("Login")
+                signup_btn = gr.Button("Sign Up")
+                logout_btn = gr.Button("Logout")
+            auth_status = gr.Textbox(label="Auth Status", interactive=False)
+
         # 📈 Prediction Tab
         with gr.Tab("📈 Prediction"):
             gr.Markdown("## 📊 Stock Prediction with LSTM & Sentiment")
@@ -718,7 +790,6 @@ with gr.Blocks(css=custom_css) as app:
             sentiment_chart = gr.Image(label="🧠 Sentiment Score Comparison", height=250)
             shap_img = gr.Image(label="🧠 Feature Influence (Approximate)", height=250)
 
-
             with gr.Row():
                 stock_news = gr.Textbox(label="📰 Stock Headlines", lines=4)
                 industry_news = gr.Textbox(label="🏭 Industry Headlines", lines=4)
@@ -738,23 +809,23 @@ with gr.Blocks(css=custom_css) as app:
             rec_btn = gr.Button("🧠 Personalized Advice")
             advice_box = gr.Textbox(label="📑 AI-Powered Investment Recommendation", lines=25, show_copy_button=True)
 
-        # 🧠 Market Mind Reader Tab
+        # 🧠 Market Mind Reader
         with gr.Tab("🧠 Market Mind Reader"):
             gr.Markdown("### 🧠 Predict Retail Investor Behavior Using Sentiment + Fundamentals")
             mm_symbol = gr.Textbox(label="Enter Stock Symbol", value="RELIANCE.NS")
             mm_btn = gr.Button("🔍 Analyze Behavior")
             mm_output = gr.Textbox(label="🧠 Behavior Prediction", lines=10, show_copy_button=True)
 
-        # 🧭 Greed & Fear Gauge Tab
+        # 🧭 Greed & Fear Gauge
         with gr.Tab("🧭 Greed & Fear Gauge"):
             gr.Markdown("### 🧭 Real-Time Greed & Fear Score")
             gr.HTML("""
-           <div style='font-size:15px;'>
-        🟥 <span style='color:red; font-weight:bold;'>Extreme Greed</span> > 70  
-        🟨 <span style='color:orange; font-weight:bold;'>Neutral/Caution</span> 30–70  
-        🟩 <span style='color:green; font-weight:bold;'>Fear/Panic</span> < 30
-          </div>
-          """)
+            <div style='font-size:15px;'>
+            🟥 <span style='color:red; font-weight:bold;'>Extreme Greed</span> > 70  
+            🟨 <span style='color:orange; font-weight:bold;'>Neutral/Caution</span> 30–70  
+            🟩 <span style='color:green; font-weight:bold;'>Fear/Panic</span> < 30
+            </div>
+            """)
             gf_symbol = gr.Textbox(label="Enter Stock Symbol", value="RELIANCE.NS")
             gf_btn = gr.Button("📊 Show Market Mood")
             gf_output = gr.Textbox(label="📉 Greed & Fear Result", lines=5, show_copy_button=True)
@@ -840,31 +911,28 @@ Made with ❤️ for investors who want smarter decisions.
             ⚠️ <strong>Disclaimer</strong>: This application is for educational purposes only. All AI-based predictions and insights are illustrative and should not be considered financial advice. Always consult a licensed financial advisor before investing.
             </div>
             """)
+        with gr.Tab("📜 My History"):
+            gr.Markdown("### 📜 View Your Saved History")
+            load_history_btn = gr.Button("Load My History")
+            history_output = gr.Textbox(label="Your History", lines=12, interactive=False)
 
     # 🔘 Button Bindings
-    refresh_btn.click(fetch_live_snapshot, inputs=symbol, outputs=[
-        live_price, day_range, rsi_val, pe_val, bb_range, week_52
-    ])
-
+    login_btn.click(fn=login_user, inputs=[email_input, password_input], outputs=[None, auth_status])
+    signup_btn.click(fn=signup_user, inputs=[email_input, password_input], outputs=auth_status)
+    logout_btn.click(fn=logout_user, outputs=auth_status)
     predict_btn.click(predict_price, inputs=symbol, outputs=[
-        pred, reco, reason,
-        chart1, chart2, volume_chart, sentiment_chart,
-        stock_news, industry_news, finance,
-        tech_box, corp_actions_box, sector_chart,
-        shap_img 
+        pred, reco, reason, chart1, chart2, volume_chart, sentiment_chart,
+        stock_news, industry_news, finance, tech_box, corp_actions_box, sector_chart, shap_img
     ])
-
-    simulate_btn.click(simulate_portfolio, inputs=portfolio_input, outputs=[
-        port_summary, port_chart
+    refresh_btn.click(fetch_live_snapshot, inputs=symbol, outputs=[
+        live_price, day_range, rsi_val, pe_val, bb_range, week_52
     ])
-
+    simulate_btn.click(simulate_portfolio, inputs=portfolio_input, outputs=[port_summary, port_chart])
     rec_btn.click(generate_recommendation, inputs=port_summary, outputs=advice_box)
-
     mm_btn.click(market_mind_reader, inputs=mm_symbol, outputs=mm_output)
     gf_btn.click(greed_fear_gauge, inputs=gf_symbol, outputs=gf_output)
-    
+    load_history_btn.click(fn=show_history, outputs=history_output)
 
-# 🚀 Launch App
 app.launch()