Update app.py

app.py

@@ -1,32 +1,6 @@
-# app.py - USING INDIVIDUAL SECRETS
+# app.py - UPDATED VERSION
 import streamlit as st
 import os
-import sys
-
-# Debug info
-print("=== APP START ===", file=sys.stderr)
-print(f"All env vars: {list(os.environ.keys())}", file=sys.stderr)
-
-def get_secret(key, default=""):
-    """Get secret from environment or Streamlit secrets"""
-    # Try environment first (HuggingFace individual secrets)
-    env_value = os.getenv(key)
-    if env_value:
-        return env_value
-    
-    # Try Streamlit secrets as fallback
-    try:
-        # For nested keys like google.client_id
-        if '.' in key:
-            parts = key.split('.')
-            value = st.secrets
-            for part in parts:
-                value = value[part]
-            return value
-        else:
-            return st.secrets[key]
-    except:
-        return default
 
 def main():
     st.set_page_config(
@@ -35,54 +9,61 @@ def main():
         layout="wide"
     )
     
-    # Get all secrets
-    client_id = get_secret("GOOGLE_CLIENT_ID")
-    client_secret = get_secret("GOOGLE_CLIENT_SECRET")
-    session_key = get_secret("SESSION_SECRET_KEY")
-    hf_token = get_secret("HUGGINGFACE_TOKEN")
-    admin_email = get_secret("ADMIN_EMAIL")
+    # Check if user is already logged in
+    if "authenticated" in st.session_state and st.session_state.authenticated:
+        show_dashboard()
+    else:
+        show_login_page()
+
+def show_dashboard():
+    """Show main dashboard after login"""
+    user = st.session_state.user_info
+    
+    # Welcome message
+    st.title(f"👋 Welcome, {user.get('name', 'User')}!")
+    st.write(f"Email: {user.get('email')}")
+    
+    # Your original dashboard content here
+    st.markdown("## 🚀 Launch Extractors")
+    
+    col1, col2, col3 = st.columns(3)
     
-    # Debug display (safe)
-    st.sidebar.write("### 🔧 Debug Info")
-    st.sidebar.write(f"Client ID loaded: {'✅' if client_id else '❌'}")
-    st.sidebar.write(f"Session key: {'✅' if session_key else '❌'}")
+    with col1:
+        if st.button("💼 LinkedIn Extractor", use_container_width=True):
+            st.switch_page("pages/linkedin_extractor.py")
     
-    # Check if essential secrets are loaded
-    if not client_id or client_id == "your-client-id.apps.googleusercontent.com":
-        st.error("""
-        ⚠️ **Google OAuth Not Configured**
-        
-        **Please add these secrets to HuggingFace:**
-        
-        1. Go to Space Settings → Repository Secrets
-        2. Add these individual secrets:
-        
-        | Secret Name | Value |
-        |-------------|-------|
-        | `GOOGLE_CLIENT_ID` | Your Google Client ID |
-        | `GOOGLE_CLIENT_SECRET` | Your Google Client Secret |
-        | `SESSION_SECRET_KEY` | `LXr9Q2v8bPp1Yt6zMk5w7x0aS3dFg4hJ7n8cV2bN5m8q9T1y3U6i0o` |
-        | `HUGGINGFACE_TOKEN` | Your HuggingFace token |
-        
-        3. Click "Save"
-        4. Restart the Space
-        """)
+    with col2:
+        if st.button("📘 Facebook Extractor", use_container_width=True):
+            st.switch_page("pages/facebook_extractor.py")
+    
+    with col3:
+        if st.button("🔥 Facebook Extractor 2.0", use_container_width=True):
+            st.switch_page("pages/facebook_extractor_pro.py")
+    
+    # Logout button
+    if st.button("🚪 Logout"):
+        st.session_state.clear()
+        st.rerun()
+
+def show_login_page():
+    """Show login page"""
+    GOOGLE_CLIENT_ID = os.getenv("GOOGLE_CLIENT_ID", "")
+    
+    if not GOOGLE_CLIENT_ID:
+        st.error("Google OAuth not configured")
         return
     
-    # Show login page
     st.title("🔐 Social Media Data Extractor")
-    st.markdown("---")
     st.markdown("### Welcome! Please login to continue")
     
     # Create login URL
     redirect_uri = "https://refat81-social-media-data-extractor-chatbot.hf.space/oauth_callback"
-    login_url = f"https://accounts.google.com/o/oauth2/v2/auth?client_id={client_id}&redirect_uri={redirect_uri}&response_type=code&scope=openid%20email%20profile&access_type=offline&prompt=consent"
+    login_url = f"https://accounts.google.com/o/oauth2/v2/auth?client_id={GOOGLE_CLIENT_ID}&redirect_uri={redirect_uri}&response_type=code&scope=openid%20email%20profile&access_type=offline&prompt=consent"
     
-    # Google login button
     st.markdown(f"""
     <a href="{login_url}">
         <div style="
-            background: linear-gradient(135deg, #4285F4, #34A853);
+            background-color: #4285F4;
             color: white;
             padding: 15px 30px;
             border-radius: 8px;
@@ -93,38 +74,12 @@ def main():
             text-align: center;
             margin: 30px 0;
             display: inline-block;
-            box-shadow: 0 4px 15px rgba(66, 133, 244, 0.3);
-            transition: all 0.3s ease;
-        "
-        onmouseover="this.style.transform='translateY(-2px)'; this.style.boxShadow='0 6px 20px rgba(66, 133, 244, 0.4)';"
-        onmouseout="this.style.transform='translateY(0)'; this.style.boxShadow='0 4px 15px rgba(66, 133, 244, 0.3)';"
-        >
+        ">
             <img src="https://cdn-icons-png.flaticon.com/512/2991/2991148.png" width="24" height="24" style="vertical-align: middle; margin-right: 12px;">
             Sign in with Google
         </div>
     </a>
     """, unsafe_allow_html=True)
-    
-    # Instructions
-    with st.expander("📋 Setup Instructions", expanded=True):
-        st.markdown(f"""
-        **Secrets Status:**
-        - Google Client ID: {'✅ Loaded' if client_id and 'placeholder' not in client_id else '❌ Not configured'}
-        - Session Key: {'✅ Loaded' if session_key else '❌ Missing'}
-        
-        **Next Steps:**
-        1. Click "Sign in with Google" above
-        2. Login with your Google account
-        3. Grant permissions to the app
-        4. You'll be redirected back to the dashboard
-        
-        **Note:** First login may show "App not verified" warning.
-        Click "Advanced" → "Go to app" to continue.
-        """)
-    
-    # Footer
-    st.markdown("---")
-    st.caption("© 2024 Social Media Data Extractor | Secure OAuth Authentication")
 
 if __name__ == "__main__":
     main()