Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-import streamlit.components.v1 as components # <<< جديد: لاستقبال الرسائل
+import streamlit.components.v1 as components # لاستقبال الرسائل
 from PIL import Image
 import torch
 from torchvision import transforms
@@ -55,10 +55,10 @@ window.addEventListener('message', (event) => {
 # ستقوم هذه الدالة بإرجاع القيمة (authenticated: true) عند نجاح الاستماع
 auth_result = components.html(auth_listener_html, height=0, width=0)
 
+# --- <<< هذا هو السطر الذي تم تصحيحه ---
 # إذا أرسل المستمع نتيجة إيجابية، قم بتحديث حالة الجلسة
-if auth_result and auth_result.get("authenticated"):
+if auth_result and "authenticated" in auth_result and auth_result["authenticated"]:
     st.session_state.authenticated = True
-
 # --- نهاية خطوة المصادقة ---
 
 
@@ -66,14 +66,63 @@ if auth_result and auth_result.get("authenticated"):
 if st.session_state.authenticated:
     
     # --- 2. كود التصميم (CSS) ---
-    # (نفس كود CSS الذي كان لديك، تم نقله إلى هنا ليعمل فقط بعد النجاح)
     CUSTOM_CSS = """
     <style>
         /* ... (نفس كود CSS بالكامل) ... */
         body, [data-testid="stAppViewContainer"], [data-testid="stHeader"] {
-            background-color: #121212 !important; 
+            background-color: #121212 !important; color: #e0e0e0 !important;
+        }
+        [data-testid="stHeader"] {
+            background-color: #1e1e1e !important; border-bottom: 1px solid #333;
+        }
+        h1, h2, h3, h4, h5, h6 { color: #ffffff !important; }
+        p, label, [data-testid="stMarkdownContainer"] { color: #e0e0e0 !important; }
+
+        /* ... (باقي كود CSS الخاص بك) ... */
+        [data-testid="stSidebar"][aria-expanded="true"] {
+            background-color: #1e1e1e !important; border-right: 1px solid #333 !important;
+        }
+        [data-testid="stSidebar"][aria-expanded="true"] [data-testid="stRadio"] label { 
+            background-color: #2a2a2a; padding: 10px; border-radius: 5px;
+            margin: 5px 0; transition: all 0.3s ease;
+        }
+        [data-testid="stSidebar"][aria-expanded="true"] [data-testid="stRadio"] label:hover {
+            background-color: #333; color: #007bff;
+        }
+        [data-testid="stSidebar"][aria-expanded="true"] [data-testid="stRadio"] input:checked + div {
+            color: #007bff; 
+        }
+        [data-testid="stDownloadButton"] > button {
+            background-color: #007bff !important; color: #ffffff !important;
+            border: none !important; border-radius: 5px !important;
+            font-weight: bold !important; padding: 12px 20px !important;
+            transition: all 0.3s ease !important; width: 100%; 
+        }
+        [data-testid="stDownloadButton"] > button:hover {
+            background-color: #0056b3 !important; transform: scale(1.03) !important;
+        }
+        [data-testid="stFileUploader"] section {
+            background-color: #1e1e1e !important; border: 1px dashed #333 !important;
+            border-radius: 8px !important;
+        }
+        [data-testid="stFileUploader"] section > button { 
+            background-color: #007bff !important; color: #ffffff !important;
+            border: none !important; border-radius: 5px !important;
+            font-weight: bold !important; transition: all 0.3s ease !important;
+        }
+        [data-testid="stFileUploader"] section > button:hover {
+            background-color: #0056b3 !important; transform: scale(1.03) !important;
+        }
+        [data-testid="stFileUploader"] label { color: #ffffff !important; }
+        [data-testid="stFileUploader"] section [data-testid="stMarkdownContainer"] p {
+            color: #aaa !important;
+        }
+        [data-testid="stImage"] figcaption { color: #aaa !important; }
+        [data-testid="stSpinner"] > div { color: #e0e0e0 !important; }
+        [data-testid="stTextInput"] input { 
+            background-color: #1e1e1e !important; color: #e0e0e0 !important;
+            border: 1px solid #333 !important; border-radius: 5px !important;
         }
-        /* ... إلخ ... */
     </style>
     """
     st.markdown(CUSTOM_CSS, unsafe_allow_html=True)
@@ -164,9 +213,6 @@ if st.session_state.authenticated:
     selected_tab = st.sidebar.radio(loc['input_method'], [loc['tab_upload'], loc['tab_url'], loc['tab_file']])
 
     # --- 8. منطق التبويبات ---
-    # (هنا نضع كل منطق التبويبات الثلاثة: tab_upload, tab_url, tab_file)
-    # ... (الكود الخاص بك كما هو) ...
-
     if selected_tab == loc['tab_upload']:
         uploaded_file = st.file_uploader(loc['upload_prompt'], type=["jpg", "jpeg", "png"])
         if uploaded_file is not None:
@@ -220,7 +266,7 @@ if st.session_state.authenticated:
             except Exception as e:
                 st.error(f"{loc['generic_error']} {e}")
 
-# --- <<< جديد: إذا فشلت المصادقة ---
+# --- إذا فشلت المصادقة ---
 else:
     # إذا لم تنجح المصادقة (st.session_state.authenticated بقيت False)
     # سيعرض رسالة الخطأ هذه بدلاً من التطبيق