Update app.py

app.py

@@ -1,147 +1,77 @@
 import streamlit as st
-import os
-import requests
-from dotenv import load_dotenv
-import bcrypt
+from datetime import datetime
 
-# Load environment variables
-load_dotenv()
-
-# Retrieve Hugging Face API key from .env file
-api_key = os.getenv("HUGGINGFACE_API_KEY")
-headers = {"Authorization": f"Bearer {api_key}"}
-
-# Function to query the Llama model using Hugging Face Inference API
-def query_llama(prompt):
-    # Replace with the correct endpoint for your open source Llama model
-    API_URL = "https://api-inference.huggingface.co/models/meta-llama/Llama-2-70b-chat"
-    payload = {"inputs": prompt}
-    response = requests.post(API_URL, headers=headers, json=payload)
-    if response.status_code == 200:
-        output = response.json()[0].get("generated_text", "No output generated.")
-        return output
-    else:
-        return f"Error: {response.status_code} - {response.text}"
+# Initialize session state for users
+if 'users' not in st.session_state:
+    st.session_state.users = {}
+if 'current_user' not in st.session_state:
+    st.session_state.current_user = None
 
-# -----------------------
-# User Authentication Setup
-# -----------------------
-if "users" not in st.session_state:
-    # Pre-register a user "kashif" with password "password" for demonstration
-    hashed = bcrypt.hashpw("password".encode(), bcrypt.gensalt())
-    st.session_state.users = {
-        "kashif": {"password": hashed, "balance": 1000.0, "transactions": []}
-    }
+# Authentication
+st.sidebar.title("🔐 User Authentication")
+username = st.sidebar.text_input("Enter Username")
+password = st.sidebar.text_input("Enter Password", type="password")
 
-if "logged_in" not in st.session_state:
-    st.session_state.logged_in = False
-if "current_user" not in st.session_state:
-    st.session_state.current_user = None
+def authenticate(username, password):
+    return username in st.session_state.users and st.session_state.users[username]['password'] == password
 
 def register_user(username, password):
-    users = st.session_state.users
-    if username in users:
-        return False, "Username already exists."
-    hashed = bcrypt.hashpw(password.encode(), bcrypt.gensalt())
-    users[username] = {"password": hashed, "balance": 0.0, "transactions": []}
-    st.session_state.users = users
-    return True, "Registration successful! Please login."
-
-def login(username, password):
-    users = st.session_state.users
-    if username in users:
-        stored_hash = users[username]["password"]
-        if bcrypt.checkpw(password.encode(), stored_hash):
-            st.session_state.logged_in = True
-            st.session_state.current_user = username
-            return True
+    if username not in st.session_state.users:
+        st.session_state.users[username] = {
+            'password': password,
+            'balance': 10000.0,
+            'transactions': []
+        }
+        return True
     return False
 
-def logout():
-    st.session_state.logged_in = False
-    st.session_state.current_user = None
-
-def update_balance(username, amount, operation):
-    if operation == "deposit":
-        st.session_state.users[username]["balance"] += amount
-    elif operation == "withdraw":
-        st.session_state.users[username]["balance"] -= amount
-
-def add_transaction(username, trans_type, amount):
-    transaction = {"type": trans_type, "amount": amount}
-    st.session_state.users[username]["transactions"].append(transaction)
-
-# -----------------------
-# Streamlit App UI
-# -----------------------
-st.title("Banking Chatbot with Llama AI Integration")
-
-# Toggle between Login and Register
-auth_mode = st.radio("Select Option", ["Login", "Register"])
-
-if auth_mode == "Register":
-    st.header("Register")
-    reg_username = st.text_input("Choose a Username", key="reg_username")
-    reg_password = st.text_input("Choose a Password", type="password", key="reg_password")
-    reg_confirm = st.text_input("Confirm Password", type="password", key="reg_confirm")
-    if st.button("Register"):
-        if reg_password != reg_confirm:
-            st.error("Passwords do not match.")
-        else:
-            success, message = register_user(reg_username, reg_password)
-            if success:
-                st.success(message)
-            else:
-                st.error(message)
-                
-if auth_mode == "Login":
-    st.header("Login")
-    username = st.text_input("Username", key="login_username")
-    password = st.text_input("Password", type="password", key="login_password")
-    if st.button("Login"):
-        if login(username, password):
-            st.success("Logged in successfully!")
-        else:
-            st.error("Invalid username or password.")
-
-if st.session_state.logged_in:
-    st.sidebar.write(f"Logged in as: {st.session_state.current_user}")
-    if st.sidebar.button("Logout"):
-        logout()
-        st.sidebar.success("Logged out.")
-
-    st.header("Banking Operations")
-    current_user = st.session_state.current_user
-    user_data = st.session_state.users[current_user]
-    st.write("**Current Balance:** $", user_data["balance"])
-
-    action = st.selectbox("Select Operation", ["Deposit", "Withdraw"])
-    amount = st.number_input("Enter Amount", min_value=0.0, step=10.0)
-    if st.button("Submit Transaction"):
-        if action == "Deposit":
-            update_balance(current_user, amount, "deposit")
-            add_transaction(current_user, "deposit", amount)
-            st.success(f"Deposited ${amount}. New balance: ${user_data['balance']}")
-        elif action == "Withdraw":
-            if user_data["balance"] >= amount:
-                update_balance(current_user, amount, "withdraw")
-                add_transaction(current_user, "withdraw", amount)
-                st.success(f"Withdrew ${amount}. New balance: ${user_data['balance']}")
-            else:
-                st.error("Insufficient balance.")
-
-    st.header("Transaction History")
-    if user_data["transactions"]:
-        for idx, t in enumerate(user_data["transactions"], start=1):
-            st.write(f"{idx}. **{t['type'].capitalize()}**: ${t['amount']}")
+if st.sidebar.button("Login"):
+    if authenticate(username, password):
+        st.session_state.current_user = username
+        st.sidebar.success("✅ Login Successful!")
     else:
-        st.write("No transactions recorded.")
+        st.sidebar.error("❌ Invalid Username or Password")
 
-    st.header("Ask Llama AI")
-    prompt = st.text_input("Enter your query for Llama:")
-    if st.button("Get AI Response"):
-        if prompt:
-            ai_response = query_llama(prompt)
-            st.write("Llama says:", ai_response)
+if st.sidebar.button("Register"):
+    if register_user(username, password):
+        st.sidebar.success("✅ Registration Successful! Please Login.")
+    else:
+        st.sidebar.error("❌ User already exists!")
+
+if st.session_state.current_user:
+    user_data = st.session_state.users[st.session_state.current_user]
+    st.title(f"🏦 Welcome, {st.session_state.current_user}!")
+    
+    # Balance
+    st.subheader("💰 Account Balance")
+    st.write(f"Current Balance: **{user_data['balance']} USD**")
+    
+    # Deposit
+    deposit_amount = st.number_input("Enter amount to deposit", min_value=0.0)
+    if st.button("Deposit"):
+        user_data['balance'] += deposit_amount
+        user_data['transactions'].append((datetime.now(), f"Deposited {deposit_amount} USD"))
+        st.success("Deposit Successful!")
+    
+    # Withdraw
+    withdraw_amount = st.number_input("Enter amount to withdraw", min_value=0.0)
+    if st.button("Withdraw"):
+        if withdraw_amount <= user_data['balance']:
+            user_data['balance'] -= withdraw_amount
+            user_data['transactions'].append((datetime.now(), f"Withdrew {withdraw_amount} USD"))
+            st.success("Withdrawal Successful!")
         else:
-            st.error("Please enter a query.")
+            st.error("Insufficient Balance!")
+    
+    # Transaction History
+    st.subheader("📜 Transaction History")
+    if user_data['transactions']:
+        for transaction in user_data['transactions']:
+            st.write(f"{transaction[0].strftime('%Y-%m-%d %H:%M:%S')}: {transaction[1]}")
+    else:
+        st.write("No transactions yet.")
+    
+    # Logout
+    if st.button("🚪 Logout"):
+        st.session_state.current_user = None
+        st.rerun()