Update app.py

app.py

@@ -8,11 +8,14 @@ import firebase_admin
 from firebase_admin import credentials, firestore
 
 # --- PAGE CONFIG ---
-st.set_page_config(page_title="Vivara Command v2", page_icon="🦁", layout="wide")
-st.title("🦁 Vivara Ecosystem Command")
+st.set_page_config(page_title="Vivara Forge", page_icon="🦁", layout="wide")
 
-# --- 1. FIREBASE CONNECTION ---
-# Checks if connected, otherwise loads from Secrets
+# --- SESSION STATE SETUP ---
+if 'is_creator' not in st.session_state:
+    st.session_state['is_creator'] = False
+
+# --- 1. FIREBASE CONNECTION (Backend) ---
+# This allows the app to talk to your database
 if not firebase_admin._apps:
     key_json = os.getenv("FIREBASE_SERVICE_ACCOUNT")
     if key_json:
@@ -20,202 +23,178 @@ if not firebase_admin._apps:
             cred_dict = json.loads(key_json)
             cred = credentials.Certificate(cred_dict)
             firebase_admin.initialize_app(cred)
-            st.toast("🔥 Firebase Connected (Admin Mode)", icon="✅")
         except Exception as e:
-            st.error(f"Firebase Key Error: {e}")
+            st.warning("⚠️ Firebase Key Error. Admin features disabled.")
     else:
-        st.warning("⚠️ Add 'FIREBASE_SERVICE_ACCOUNT' secret to enable User Management.")
+        st.warning("⚠️ Admin Database Not Connected (Add FIREBASE_SERVICE_ACCOUNT secret)")
 
 try:
     db = firestore.client()
 except:
     db = None
 
-# --- SIDEBAR CONFIG ---
+# --- SIDEBAR: LOGIN & WALLET ---
 with st.sidebar:
-    st.header("⚙️ Settings")
+    st.header("👤 Identity")
+    
+    # LOGIN FORM
+    if not st.session_state['is_creator']:
+        with st.expander("Creator Login"):
+            admin_pass = st.text_input("Admin Password", type="password")
+            if st.button("Unlock Creator Mode"):
+                # Check against the secret you saved in Hugging Face
+                if admin_pass == os.getenv("ADMIN_PASSWORD"):
+                    st.session_state['is_creator'] = True
+                    st.success("Welcome back, Fugth.")
+                    st.rerun()
+                else:
+                    st.error("Wrong Password")
+    else:
+        st.success("🟢 CREATOR MODE ACTIVE")
+        if st.button("Logout"):
+            st.session_state['is_creator'] = False
+            st.rerun()
+
+    st.divider()
+    
+    st.header("⚙️ Network Settings")
     rpc_url = st.text_input("RPC URL", value="https://polygon-rpc.com")
     chain_id = st.number_input("Chain ID", value=137)
     
-    private_key = os.getenv("PRIVATE_KEY")
-    if private_key:
-        st.success("🔑 Wallet Key Loaded")
+    # Wallet Status
+    pk = os.getenv("PRIVATE_KEY")
+    if pk:
+        st.info("🔑 Admin Wallet Loaded")
     else:
-        st.error("⚠️ Add 'PRIVATE_KEY' secret")
+        st.warning("⚠️ No Wallet Key Found")
 
-# --- TABS ---
-tab1, tab2, tab3 = st.tabs(["👥 User Manager", "📧 Newsletter & Growth", "⚒️ Solidity Forge"])
+# --- MAIN APP LOGIC ---
 
-# ==========================================
-# TAB 1: USER MANAGER (TIERS)
-# ==========================================
-with tab1:
-    st.header("User Database")
-    
-    if db:
-        users_ref = db.collection("users")
-        all_users = list(users_ref.stream())
-        
-        # Calculate Stats
-        total_users = len(all_users)
-        founders = 0
-        alphas = 0
-        betas = 0
-        
-        user_data = []
-        for doc in all_users:
-            u = doc.to_dict()
-            inv = u.get("inventory", [])
-            rank = u.get("redeem_rank", "-")
-            
-            status = "User"
-            if "FOUNDER" in inv: 
-                founders += 1
-                status = "🏆 GOLD"
-            elif "ALPHA" in inv:
-                alphas += 1
-                status = "🥈 SILVER"
-            elif "BETA" in inv:
-                betas += 1
-                status = "🥉 BRONZE"
-                
-            user_data.append({
-                "ID": doc.id[:6],
-                "Name": u.get("displayName", "Unknown"),
-                "Email": u.get("email", "-"),
-                "Status": status,
-                "Rank": rank
-            })
+# IF CREATOR: Show Everything (3 Tabs)
+# IF PUBLIC: Show Only Forge (1 Tab)
 
-        # Display Metrics
-        c1, c2, c3, c4 = st.columns(4)
-        c1.metric("Total Users", total_users)
-        c2.metric("🏆 Gold Founders", founders)
-        c3.metric("🥈 Silver Alphas", alphas)
-        c4.metric("Revenue (Est)", f"${(founders + alphas + betas) * 5}")
-        
-        st.divider()
-        
-        # User Table
-        st.subheader("User List")
-        df_users = pd.DataFrame(user_data)
-        st.dataframe(df_users, use_container_width=True)
-        
-        # Manual Grant Tool
-        st.divider()
-        st.subheader("🛠️ Manual Grant (Bypass Payment)")
-        with st.form("grant_form"):
-            c_email, c_tier = st.columns([2, 1])
-            target_email = c_email.text_input("User Email")
-            tier_select = c_tier.selectbox("Tier", ["FOUNDER", "ALPHA", "BETA"])
-            
-            if st.form_submit_button("Grant Access"):
-                # Search for user
-                q = users_ref.where("email", "==", target_email).limit(1).get()
-                if q:
-                    target_doc = q[0].reference
-                    # Update Inventory
-                    target_doc.update({
-                        "inventory": firestore.ArrayUnion([tier_select, "LICENSE_V1", "MANUAL_GRANT"]),
-                        "redeem_rank": 0 # Manual grants get Rank 0
-                    })
-                    st.success(f"Granted {tier_select} to {target_email}")
-                else:
-                    st.error("User email not found. Tell them to sign up first.")
+if st.session_state['is_creator']:
+    st.title("🦁 Vivara Command Center")
+    tabs = st.tabs(["⚒️ Solidity Forge", "👥 User Database", "💰 Coin & Stats"])
+else:
+    st.title("⚒️ Vivara Forge (Public)")
+    st.caption("Open Source Smart Contract Deployment Tool")
+    tabs = st.tabs(["⚒️ Solidity Forge"])
 
 # ==========================================
-# TAB 2: NEWSLETTER (MARKETING)
+# TAB 1: SOLIDITY FORGE (PUBLIC)
 # ==========================================
-with tab2:
-    st.header("📧 Email List")
+with tabs[0]:
+    col1, col2 = st.columns([1.5, 1])
     
-    if db:
-        news_ref = db.collection("newsletters")
-        emails = list(news_ref.order_by("timestamp", direction=firestore.Query.DESCENDING).stream())
-        
-        st.metric("Total Subscribers", len(emails))
-        
-        if len(emails) > 0:
-            email_data = []
-            for doc in emails:
-                d = doc.to_dict()
-                email_data.append({
-                    "Email": d.get("email"),
-                    "Date": d.get("timestamp"),
-                    "Source": d.get("source", "unknown")
-                })
-            
-            df_emails = pd.DataFrame(email_data)
-            st.dataframe(df_emails, use_container_width=True)
-            
-            # Export Button
-            csv = df_emails.to_csv(index=False).encode('utf-8')
-            st.download_button(
-                "📥 Download Email List (CSV)",
-                csv,
-                "vivara_emails.csv",
-                "text/csv",
-                key='download-csv'
-            )
-        else:
-            st.info("No subscribers yet.")
-
-# ==========================================
-# TAB 3: SOLIDITY FORGE (CONTRACTS)
-# ==========================================
-with tab3:
-    col_a, col_b = st.columns([1.5, 1])
-    
-    with col_a:
+    with col1:
         st.subheader("📝 Contract Editor")
         default_code = """// SPDX-License-Identifier: MIT
 pragma solidity ^0.8.20;
 import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
 
-contract Vivara is ERC20 {
-    constructor() ERC20("Vivara", "VIVR") {
-        _mint(msg.sender, 1000000000 * 10**18);
+contract MyToken is ERC20 {
+    constructor() ERC20("MyToken", "MTK") {
+        _mint(msg.sender, 1000 * 10**18);
     }
 }
 """
         code_input = st.text_area("Solidity Code", value=default_code, height=500)
 
-    with col_b:
-        st.subheader("🚀 Actions")
-        
+    with col2:
+        st.subheader("🚀 Compiler")
         if st.button("COMPILE"):
-            with st.spinner("Compiling..."):
+            with st.spinner("Installing Compiler..."):
                 try:
                     install_solc('0.8.20')
-                    compiled_sol = compile_source(code_input, output_values=['abi', 'bin'], solc_version='0.8.20')
-                    contract_id, contract_interface = list(compiled_sol.items())[-1]
-                    st.session_state['abi'] = contract_interface['abi']
-                    st.session_state['bytecode'] = contract_interface['bin']
-                    st.success(f"Compiled: {contract_id}")
+                    compiled = compile_source(code_input, output_values=['abi', 'bin'], solc_version='0.8.20')
+                    contract_id, iface = list(compiled.items())[-1]
+                    st.session_state['abi'] = iface['abi']
+                    st.session_state['bin'] = iface['bin']
+                    st.success(f"✅ Compiled: {contract_id}")
                 except Exception as e:
                     st.error(f"Error: {e}")
+        
+        st.divider()
+        
+        # DEPLOY BUTTON (Only enabled if Creator logged in OR use Metamask injection trick)
+        # Since this is Python-based, usually only the Admin deploys via the server key.
+        if st.session_state['is_creator']:
+            if st.button("DEPLOY (Admin Wallet)"):
+                if not pk:
+                    st.error("No Private Key!")
+                else:
+                    try:
+                        w3 = Web3(Web3.HTTPProvider(rpc_url))
+                        acct = w3.eth.account.from_key(pk)
+                        Contract = w3.eth.contract(abi=st.session_state['abi'], bytecode=st.session_state['bin'])
+                        tx = Contract.constructor().build_transaction({
+                            'from': acct.address,
+                            'nonce': w3.eth.get_transaction_count(acct.address),
+                            'gas': 3000000,
+                            'gasPrice': w3.to_wei('50', 'gwei'),
+                            'chainId': int(chain_id)
+                        })
+                        signed = w3.eth.account.sign_transaction(tx, pk)
+                        tx_hash = w3.eth.send_raw_transaction(signed.rawTransaction)
+                        st.success(f"🚀 Deployed! TX: {w3.to_hex(tx_hash)}")
+                    except Exception as e:
+                        st.error(f"Deploy Error: {e}")
+        else:
+            st.info("🔒 Server-side deployment is restricted to Admins. Public deployment via MetaMask coming soon.")
 
-        if st.button("DEPLOY TO POLYGON"):
-            if 'bytecode' not in st.session_state:
-                st.warning("Compile first!")
-            elif not private_key:
-                st.error("No Private Key!")
-            else:
-                try:
-                    w3 = Web3(Web3.HTTPProvider(rpc_url))
-                    acct = w3.eth.account.from_key(private_key)
-                    
-                    Contract = w3.eth.contract(abi=st.session_state['abi'], bytecode=st.session_state['bytecode'])
-                    construct_txn = Contract.constructor().build_transaction({
-                        'from': acct.address,
-                        'nonce': w3.eth.get_transaction_count(acct.address),
-                        'gas': 3000000,
-                        'gasPrice': w3.to_wei('50', 'gwei'),
-                        'chainId': int(chain_id)
-                    })
+# ==========================================
+# TAB 2: USER DATABASE (CREATOR ONLY)
+# ==========================================
+if st.session_state['is_creator']:
+    with tabs[1]:
+        st.header("👥 User Management")
+        if db:
+            users_ref = db.collection("users")
+            all_users = list(users_ref.stream())
+            
+            founders = 0
+            alphas = 0
+            
+            data = []
+            for doc in all_users:
+                u = doc.to_dict()
+                inv = u.get("inventory", [])
+                
+                tier = "Free"
+                if "FOUNDER" in inv: 
+                    tier = "🏆 GOLD"
+                    founders += 1
+                elif "ALPHA" in inv: 
+                    tier = "🥈 SILVER"
+                    alphas += 1
                     
-                    signed = w3.eth.account.sign_transaction(construct_txn, private_key)
-                    tx_hash = w3.eth.send_raw_transaction(signed.rawTransaction)
-                    st.success(f"🚀 Deployed! TX: {w3.to_hex(tx_hash)}")
-                except Exception as e:
-                    st.error(f"Deploy Error: {e}")
+                data.append({
+                    "Name": u.get("displayName", "Unknown"),
+                    "Email": u.get("email", "-"),
+                    "Tier": tier,
+                    "Rank": u.get("redeem_rank", "-")
+                })
+            
+            k1, k2, k3 = st.columns(3)
+            k1.metric("Total Users", len(all_users))
+            k2.metric("Founders", founders)
+            k3.metric("Revenue", f"${(founders + alphas) * 5}")
+            
+            st.dataframe(pd.DataFrame(data), use_container_width=True)
+            
+            # NEWSLETTER EXPORT
+            st.subheader("📧 Newsletter List")
+            news_docs = list(db.collection("newsletters").stream())
+            emails = [d.to_dict().get("email") for d in news_docs]
+            st.write(f"collected {len(emails)} emails")
+            st.download_button("Download CSV", pd.DataFrame(emails, columns=["Email"]).to_csv(), "emails.csv")
+
+# ==========================================
+# TAB 3: COIN & STATS (CREATOR ONLY)
+# ==========================================
+if st.session_state['is_creator']:
+    with tabs[2]:
+        st.header("💰 Vivara Coin Stats")
+        # You can add coin burning logic here later
+        st.info("Connect Token Contract to see live burn stats.")