Create app.py

app.py

@@ -0,0 +1,80 @@
+import os
+import json
+import asyncio
+import streamlit as st
+import logging
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.FileHandler('app.log'),
+        logging.StreamHandler()
+    ]
+)
+
+# Load Google OAuth config from environment variables
+GOOGLE_OAUTH_CONFIG = {
+    "web": {
+        "client_id": os.environ.get("GOOGLE_CLIENT_ID"),
+        "project_id": os.environ.get("GOOGLE_PROJECT_ID"),
+        "auth_uri": os.environ.get("GOOGLE_AUTH_URI", "https://accounts.google.com/o/oauth2/auth"),
+        "token_uri": os.environ.get("GOOGLE_TOKEN_URI", "https://oauth2.googleapis.com/token"),
+        "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
+        "client_secret": os.environ.get("GOOGLE_CLIENT_SECRET"),
+        "redirect_uris": [os.environ.get("GOOGLE_REDIRECT_URI")]
+    }
+}
+
+# Setup the UI
+st.set_page_config(page_title="Advanced File Downloader", layout="wide", page_icon="📁")
+
+# Import the core components (still keeping modular organization)
+from utils import USER_AGENTS, STEALTH_SETTINGS, PROXY_ROTATION_CONFIG
+from utils import (
+    get_random_user_agent, sizeof_fmt, create_zip_file, humanize_file_size, get_domain, 
+    is_download_link, normalize_download_url, detect_captcha, show_user_friendly_error
+)
+from google_drive import (
+    get_google_auth_url, exchange_code_for_credentials, google_drive_upload, create_drive_folder
+)
+from download_manager import DownloadManager
+from rag_search import EnhancedRAGSearch
+
+# Initialize session state variables
+def initialize_session_state():
+    if 'files' not in st.session_state:
+        st.session_state.files = []
+    if 'downloaded_paths' not in st.session_state:
+        st.session_state.downloaded_paths = []
+    if 'download_complete' not in st.session_state:
+        st.session_state.download_complete = False
+    if 'selected_tab' not in st.session_state:
+        st.session_state.selected_tab = 0
+    if 'rag_search' not in st.session_state:
+        st.session_state.rag_search = EnhancedRAGSearch()
+    if 'keep_progress' not in st.session_state:
+        st.session_state.keep_progress = False
+    if 'google_credentials' not in st.session_state:
+        st.session_state.google_credentials = None
+    if 'mode' not in st.session_state:
+        st.session_state.mode = "Standard"
+    if 'use_proxy' not in st.session_state:
+        st.session_state.use_proxy = False
+    if 'proxy_string' not in st.session_state:
+        st.session_state.proxy_string = None
+    if 'stealth_mode' not in st.session_state:
+        st.session_state.stealth_mode = True
+
+# Import the UI code while keeping the modular structure
+from ui import setup_ui, create_sidebar, display_file_results, handle_downloads, handle_google_drive_upload
+from main import main as app_main
+
+# Set up and run the application
+def main():
+    initialize_session_state()
+    app_main()
+
+if __name__ == "__main__":
+    main()