Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 import re
 import shutil
 import json
+import secrets
 from datetime import datetime, timedelta
 from urllib.parse import unquote
 
@@ -25,12 +26,14 @@ os.makedirs("system", exist_ok=True)
 
 REPO_ID = "Asem75/aiocr_asistant"
 
-RAW_DIR = "RAW_FILES"           
+RAW_DIR = "RAW_FILES"
 WATCH_DIR_1 = "process"
 WATCH_DIR_2 = "process_files"
 
 FINAL_DIR = "Final_Arabic_Files"
 SYSTEM_STATUS_PATH = "system/status.json"
+ABOUT_APP_PATH = "system/about_app.json"     # ✅ جديد: تفاصيل الاشتراك / محتوى يكتبه المدير
+MESSAGES_PATH = "system/messages.json"       # ✅ جديد: قناة تقارير الواجهة لمكتب المدير
 
 TOKEN = os.getenv("AIOCR_KEY")
 if not TOKEN:
@@ -56,6 +59,115 @@ def sanitize_filename(name: str) -> str:
     name = name.replace(" ", "_")
     return re.sub(r"[^\w\.\-]", "", name)
 
+def generate_guest_name() -> str:
+    """
+    🧩 توليد معرّف مؤقت لكل جلسة بانتظار دمج "كود التعرف" مستقبلاً
+    (الاسم + البريد الإلكتروني + بصمة عتاد الجهاز + الرتبة).
+    عند إضافة ذلك الكود، يكفي تمرير بياناته الحقيقية إلى user_session
+    بدلاً من هذا المعرّف المؤقت، وسيُحدَّث الترحيب وأيقونة الرتبة تلقائياً.
+    """
+    return "زائر_" + secrets.token_hex(3)
+
+# ------------------------------------------------------------
+# ✅ (1) تحديث ملف status.json في الخزنة السحابية بعد كل عملية رفع
+# ------------------------------------------------------------
+def update_status_json(status_text: str):
+    try:
+        local_path = os.path.join("system", "status.json")
+        data = {
+            "status": status_text,
+            "updated": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        }
+        with open(local_path, "w", encoding="utf-8") as f:
+            json.dump(data, f, ensure_ascii=False, indent=2)
+
+        api.upload_file(
+            path_or_fileobj=local_path,
+            path_in_repo=SYSTEM_STATUS_PATH,
+            repo_id=REPO_ID,
+            repo_type="dataset",
+            token=TOKEN,
+        )
+    except Exception as e:
+        log_error(f"STATUS UPDATE ERROR: {e}")
+
+# ------------------------------------------------------------
+# ✅ (4) إرسال تقرير من الواجهة إلى مكتب المدير عبر system/messages.json
+# ------------------------------------------------------------
+def send_report_to_manager(report_text: str):
+    try:
+        try:
+            local_existing = hf_hub_download(
+                repo_id=REPO_ID,
+                filename=MESSAGES_PATH,
+                repo_type="dataset",
+                token=TOKEN,
+            )
+            with open(local_existing, "r", encoding="utf-8") as f:
+                messages = json.load(f)
+            if not isinstance(messages, list):
+                messages = []
+        except Exception:
+            messages = []
+
+        messages.append({
+            "time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            "from": "upload_interface",
+            "report": report_text,
+        })
+
+        local_path = os.path.join("system", "messages.json")
+        with open(local_path, "w", encoding="utf-8") as f:
+            json.dump(messages, f, ensure_ascii=False, indent=2)
+
+        api.upload_file(
+            path_or_fileobj=local_path,
+            path_in_repo=MESSAGES_PATH,
+            repo_id=REPO_ID,
+            repo_type="dataset",
+            token=TOKEN,
+        )
+    except Exception as e:
+        log_error(f"REPORT SEND ERROR: {e}")
+
+# ------------------------------------------------------------
+# ✅ (3) تحميل تفاصيل الاشتراك من system/about_app.json (يكتبها المدير)
+# ------------------------------------------------------------
+def load_about_app(rank: str = "free") -> str:
+    data = {}
+    try:
+        local_path = hf_hub_download(
+            repo_id=REPO_ID,
+            filename=ABOUT_APP_PATH,
+            repo_type="dataset",
+            token=TOKEN,
+        )
+        with open(local_path, "r", encoding="utf-8") as f:
+            data = json.load(f)
+    except Exception as e:
+        log_error(f"ABOUT APP READ ERROR: {e}")
+
+    if isinstance(data, dict):
+        section_text = (
+            data.get(rank)
+            or data.get("message")
+            or data.get("about")
+            or "لا تتوفر تفاصيل إضافية حالياً."
+        )
+    else:
+        section_text = "لا تتوفر تفاصيل إضافية حالياً."
+
+    badge = "🥇 عضوية VIP" if rank == "vip" else "🥈 عضوية مجانية (Free)"
+
+    html = f"""
+    <div role="dialog" aria-label="تفاصيل الاشتراك"
+         style="background:#ffffff; border:1px solid #ddd; border-radius:8px; padding:16px; margin-top:8px;">
+        <h3 style="margin-top:0;">{badge}</h3>
+        <p style="white-space:pre-line; margin:0;">{section_text}</p>
+    </div>
+    """
+    return html
+
 # ============================================================
 # 2) رادار فحص القيود للمشترك المجاني (FREE)
 # ============================================================
@@ -122,35 +234,45 @@ def read_status_json():
         with open("./system/status.json", "r", encoding="utf-8") as f:
             data = json.load(f)
 
+        status_text = data.get('status', 'غير معروف')
+        updated_text = data.get('updated', 'غير معروف')
+
+        # ✅ (2) مربع الحالة الآن متوافق مع قارئ الشاشة:
+        # role="status" + aria-live="polite" تجعل البرنامج يقرأ كل تحديث تلقائياً
         html = f"""
-        <div style="font-size:16px; background-color: #f0f7ff; padding: 10px; border-radius: 5px; border: 1px solid #b3d7ff;">
-            <b>📌 حالة نظام المعالجة الحالية:</b> {data.get('status','غير معروف')}<br>
-            <b>🕒 آخر تحديث للسجل:</b> {data.get('updated','غير معروف')}
+        <div role="status" aria-live="polite" aria-atomic="true"
+             style="font-size:16px; background-color: #f0f7ff; padding: 10px; border-radius: 5px; border: 1px solid #b3d7ff;">
+            <p style="margin:4px 0;"><b>📌 حالة نظام المعالجة الحالية:</b> <span>{status_text}</span></p>
+            <p style="margin:4px 0;"><b>🕒 آخر تحديث للسجل:</b> <span>{updated_text}</span></p>
         </div>
         """
         return html
     except Exception as e:
         log_error(f"STATUS READ ERROR: {e}")
-        return f"<div style='color:orange;'>⚠️ لا يمكن قراءة ملف status.json حالياً — {e}</div>"
+        return f"""<div role="status" aria-live="polite" style='color:orange;'>⚠️ لا يمكن قراءة ملف status.json حالياً — {e}</div>"""
 
 # ============================================================
 # 4) معالجة رفع الملفات مع دعم جميع البادئات
 # ============================================================
 
-def handle_upload(files, username, rank, 
-                  enable_table, enable_handwriting, 
-                  enable_heavy_prep, enable_high_dpi, 
+def handle_upload(files, session,
+                  enable_table, enable_handwriting,
+                  enable_heavy_prep, enable_high_dpi,
                   enable_fast, enable_tashkeel):
-    if not files or not username:
-        return "<div>❌ يرجى اختيار ملفات وكتابة اسم المستخدم بشكل صحيح</div>"
-
-    username = sanitize_filename(username.strip())
+    if not files:
+        return "<div role='alert'>❌ يرجى اختيار ملفات للرفع</div>"
+
+    session = session or {}
+    # ⚠️ مؤقتاً: الاسم يولَّد تلقائياً للجلسة. سيُستبدل باسم المستخدم الحقيقي
+    # فور دمج "كود التعرف" (الاسم + البريد + بصمة الجهاز + الرتبة).
+    username = sanitize_filename((session.get("name") or generate_guest_name()).strip())
+    rank = session.get("rank", "free")
     is_vip = (rank == "vip")
-    
+
     if not is_vip:
         allowed, message = check_free_user_limits(username)
         if not allowed:
-            return f"<div style='color:red; font-weight:bold;'>{message}</div>"
+            return f"<div role='alert' style='color:red; font-weight:bold;'>{message}</div>"
 
     success = 0
     prefix = f"VIP_{username}_" if is_vip else f"{username}_"
@@ -161,7 +283,7 @@ def handle_upload(files, username, rank,
             file_size = os.path.getsize(file.name)
 
             if not is_vip and file_size > 3 * 1024 * 1024:
-                return "<div style='color:red;'>❌ خطأ: أحد الملفات يتجاوز حجمه 3 ميجابايت!</div>"
+                return "<div role='alert' style='color:red;'>❌ خطأ: أحد الملفات يتجاوز حجمه 3 ميجابايت!</div>"
 
             # بناء بادئة الخيارات
             option_prefix = ""
@@ -200,7 +322,16 @@ def handle_upload(files, username, rank,
         except Exception as e:
             log_error(f"UPLOAD ERROR: {file.name} — {e}")
 
-    return f"<div style='color:green; font-weight:bold;'>تم رفع {success} ملف/ات مع الخيارات المحددة ✔</div>"
+    # ✅ (1) كتابة "تم رفع الملف" في system/status.json على الخزنة السحابية
+    if success > 0:
+        update_status_json("تم رفع الملف" if success == 1 else f"تم رفع {success} ملفات")
+
+    # ✅ (4) إرسال تقرير بهذه العملية إلى مكتب المدير (system/messages.json)
+    send_report_to_manager(
+        f"المستخدم {username} ({'VIP' if is_vip else 'FREE'}) رفع {success} من {len(files)} ملف/ات."
+    )
+
+    return f"<div role='status' style='color:green; font-weight:bold;'>تم رفع {success} ملف/ات مع الخيارات المحددة ✔</div>"
 
 # ============================================================
 # 5) إدارة وقراءة ملفات Final والتحويل (باستخدام hf_hub_download)
@@ -259,6 +390,31 @@ def do_zip(text, base_name):
     except Exception as e:
         return "⚠️ فشل إنشاء ZIP", None
 
+# ============================================================
+# 5.b) الجلسة الحالية: ترحيب بالاسم + أيقونة الرتبة
+#      (نقطة الدخول المستقبلية لـ "كود التعرف")
+# ============================================================
+
+def init_session():
+    """
+    🧩 يُستدعى عند تحميل الصفحة. حالياً يولّد جلسة "زائر" مؤقتة بالرتبة free.
+    عندما يُضاف "كود التعرف" (اسم + بريد + بصمة جهاز + رتبة)، يكفي استبدال
+    منطق هذه الدالة بنتيجة ذلك الكود، وسيُحدَّث الترحيب + الأيقونة + الشاشة
+    المعروضة تلقائياً بحسب الرتبة (vip / free) دون أي تغيير آخر في الواجهة.
+    """
+    session = {"name": generate_guest_name(), "email": "", "device_id": "", "rank": "free"}
+    welcome_html = f"<p role='status' style='font-size:16px; margin:0;'>👋 مرحباً بك، <b>{session['name']}</b></p>"
+    icon_label = "🥇 VIP" if session["rank"] == "vip" else "🥈 FREE"
+    return session, welcome_html, icon_label
+
+def toggle_subscription_panel(session, is_visible):
+    """فتح/إغلاق شاشة تفاصيل الاشتراك المرتبطة بـ system/about_app.json"""
+    new_visible = not is_visible
+    if new_visible:
+        rank = (session or {}).get("rank", "free")
+        return gr.update(value=load_about_app(rank), visible=True), new_visible
+    return gr.update(visible=False), new_visible
+
 # ============================================================
 # 6) واجهات Gradio مع جميع الخيارات
 # ============================================================
@@ -266,8 +422,18 @@ def do_zip(text, base_name):
 with gr.Blocks(title="📤 بوابة الرفع") as upload_page:
     gr.Markdown("<h1 style='text-align:center;'>📤 بوابة الرفع ودورة حياة الملفات</h1>")
 
-    username_input = gr.Textbox(label="👤 اسم المستخدم", value="Divid")
-    rank_input = gr.Radio(label="🏅 الرتبة", choices=["free", "vip"], value="free")
+    # ✅ (5) أُزيل مربع "اسم المستخدم" التجريبي. تُولَّد الجلسة تلقائياً وتُستبدل
+    #        مستقبلاً ببيانات "كود التعرف" الحقيقية.
+    user_session = gr.State({"name": "", "email": "", "device_id": "", "rank": "free"})
+    panel_visible_state = gr.State(False)
+
+    # ✅ (3) بدل خيار vip/free النصي: ترحيب بالاسم + أيقونة ذهبية/فضية للرتبة
+    with gr.Row():
+        welcome_box = gr.HTML("<p role='status'>👋 جارٍ تجهيز جلستك...</p>")
+        rank_icon_btn = gr.Button("🥈 FREE", elem_id="rank_icon_btn", scale=0, min_width=110)
+
+    # الشاشة العامة لتفاصيل الاشتراك (مرتبطة بـ system/about_app.json الذي يكتبه المدير)
+    subscription_panel = gr.HTML(visible=False)
 
     uploader = gr.File(label="اختر ملفاتك", file_count="multiple")
 
@@ -300,22 +466,32 @@ with gr.Blocks(title="📤 بوابة الرفع") as upload_page:
     status_box = gr.HTML("<div>بانتظار رفع الملفات...</div>")
     
     gr.Markdown("### 📜 حالة التجميع والمعالجة الفورية")
-    system_status = gr.HTML("<div>بانتظار تحديث الحالة من system/status.json...</div>")
+    system_status = gr.HTML("<div role='status' aria-live='polite'>بانتظار تحديث الحالة من system/status.json...</div>")
+
+    # تجهيز الجلسة (الترحيب + أيقونة الرتبة) عند فتح الصفحة
+    upload_page.load(init_session, None, [user_session, welcome_box, rank_icon_btn])
 
-    # ربط زر الرفع بالدالة المعدلة
+    # فتح/إغلاق شاشة تفاصيل الاشتراك عند الضغط على أيقونة الرتبة
+    rank_icon_btn.click(
+        fn=toggle_subscription_panel,
+        inputs=[user_session, panel_visible_state],
+        outputs=[subscription_panel, panel_visible_state]
+    )
+
+    # ربط زر الرفع بالدالة المعدلة (تعتمد على الجلسة بدل مربع الاسم/الرتبة)
     upload_btn.click(
         fn=handle_upload,
-        inputs=[uploader, username_input, rank_input,
+        inputs=[uploader, user_session,
                 enable_table, enable_handwriting, 
                 enable_heavy_prep, enable_high_dpi, 
                 enable_fast, enable_tashkeel],
         outputs=status_box
-    )
+    ).then(read_status_json, None, system_status)
 
     status_timer = gr.Timer(3.0)
     status_timer.tick(read_status_json, None, system_status)
 
-# ------------------- قسم ملفاتي المُصحَّح -------------------
+# ------------------- قسم ملفاتي المُصحَّح -------------------
 with gr.Blocks(title="📁 ملفاتي") as files_page:
     gr.Markdown("<h1 style='text-align:center;'>📁 الملفات الجاهزة للتحويل</h1>")
     files_list = gr.Dropdown(label="اختر ملفاً من Final_Arabic_Files", choices=[], interactive=True)
@@ -359,4 +535,4 @@ with gr.Blocks(title="📁 ملفاتي") as files_page:
 app = gr.TabbedInterface([upload_page, files_page], ["📤 الرفع المباشر", "📁 ملفاتي"])
 
 if __name__ == "__main__":
-    app.launch(server_name="0.0.0.0", server_port=7860)
+    app.launch(server_name="0.0.0.0", server_port=7860)