Update app.py

app.py

@@ -2,94 +2,50 @@ import gradio as gr
 import os
 from datetime import datetime
 from huggingface_hub import HfApi, CommitOperationAdd
-from docx import Document
-from reportlab.pdfgen import canvas
-import pandas as pd
 
-# --- [1] بروتوكول العناوين المثبت ---
+# --- الإعدادات (تأكد من مطابقة الاسم الجديد) ---
 REPO_ID = "Asem75/AiOCR_asistant" 
 TOKEN_NAME = "AIOCR_KEY"
 HF_TOKEN = os.environ.get(TOKEN_NAME)
-RAW_PATH = "raw_under_score_files"
-COMM_VAULT = "communication_vault"
 api = HfApi()
 
-# --- [2] المحركات الموفرة للرصيد (Optimized) ---
-
-def upload_and_log_optimized(key_name, file_obj, manual_text):
-    """يرسل الملف والتقرير في 'صندوق واحد' لتوفير الرصيد"""
-    if not key_name: return "⚠️ المفتاح مطلوب!"
+def fresh_start_upload(key_name, manual_text):
+    if not key_name: return "⚠️ أدخل المفتاح"
+    
+    clean_key = "".join(filter(str.isalnum, key_name))
     ts = datetime.now().strftime("%Y%m%d_%H%M%S")
-    safe_key = "".join(x for x in key_name if x.isalnum() or x in ['_', '-'])
+    
+    # بناء المسار النظيف للمستودع الجديد
+    raw_path = f"raw_under_score_files/{clean_key}_{ts}.txt"
+    log_path = f"communication_vault/LOG_{clean_key}_{ts}.txt"
     
     try:
-        operations = []
-        # تجهيز الملف الخام
-        content = manual_text.encode("utf-8") if manual_text else b"Empty Content"
-        operations.append(CommitOperationAdd(
-            path_in_repo=f"{RAW_PATH}/{safe_key}_{ts}.txt",
-            path_or_fileobj=content
-        ))
-        # تجهيز اللوج (في نفس العملية)
-        log_data = f"Action: Upload\nUser: {key_name}\nTime: {ts}".encode("utf-8")
-        operations.append(CommitOperationAdd(
-            path_in_repo=f"{COMM_VAULT}/LOG_{safe_key}_{ts}.txt",
-            path_or_fileobj=log_data
-        ))
-
-        # تنفيذ الرفع الرسمي (يستهلك 1 فقط من الرصيد)
+        content = manual_text.encode("utf-8") if manual_text else b"New Start Content"
+        
+        operations = [
+            CommitOperationAdd(path_in_repo=raw_path, path_or_fileobj=content),
+            CommitOperationAdd(path_in_repo=log_path, path_or_fileobj=f"Init: {ts}".encode("utf-8"))
+        ]
+        
         api.create_commit(
             repo_id=REPO_ID,
             operations=operations,
-            commit_message=f"Smart upload by {key_name}",
+            commit_message="Fresh start - New structure",
             repo_type="dataset",
             token=HF_TOKEN
         )
-        return "🚀 تم الرفع بنجاح! تم استخدام عملية واحدة فقط (توفير للرصيد)."
+        return f"✅ مبروك! الخزنة الجديدة استقبلت أول ملف في المجلد الصحيح."
     except Exception as e:
-        return f"❌ خطأ: {str(e)}"
-
-def get_file_list(key_name):
-    """رادار يدوي: لا يفحص إلا عند الطلب"""
-    if not key_name: return gr.update(choices=[]), "⚠️ أدخل المفتاح أولاً"
-    try:
-        files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN)
-        # فلترة الملفات الخاصة بالمستخدم فقط
-        user_files = [f for f in files if key_name in f]
-        return gr.update(choices=user_files), "🔄 تم تحديث القائمة بنجاح."
-    except:
-        return gr.update(choices=[]), "❌ فشل الاتصال بالخزنة."
-
-# --- [3] الواجهة (UI) الموفرة للطاقة ---
-
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🛡️ AiOCR Assistant V1 Pro (Smart Mode)")
-    
-    with gr.Tabs():
-        # التبويب 1: الرفع
-        with gr.TabItem("📥 بوابة الرفع", id=0):
-            with gr.Row():
-                key_in = gr.Textbox(label="🔑 المفتاح الخاص")
-                file_in = gr.File(label="📄 المستند")
-            manual_in = gr.Textbox(label="✍️ نص يدوي", lines=4)
-            up_btn = gr.Button("🚀 إرسال للخزنة", variant="primary")
-            up_stat = gr.Textbox(label="📡 الحالة")
-            up_btn.click(upload_and_log_optimized, inputs=[key_in, file_in, manual_in], outputs=up_stat)
-
-        # التبويب 2: الرادار اليدوي
-        with gr.TabItem("📂 ملفاتي", id=1):
-            with gr.Row():
-                file_list = gr.Dropdown(label="🔍 رادار الملفات (اضغط تحديث للرؤية)")
-                ref_btn = gr.Button("🔄 تحديث يدوي", scale=0)
-            editor = gr.Textbox(label="📝 المحرر", lines=12)
-            ref_btn.click(get_file_list, inputs=key_in, outputs=[file_list, up_stat])
-
-        # التبويب 3: التحويل
-        with gr.TabItem("📚 مكتبة التحويل", id=2):
-            fmt_choice = gr.Radio(["PDF", "DOCX", "XLSX", "TXT"], label="الصيغة", value="PDF")
-            go_btn = gr.Button("📩 تحميل للجهاز", variant="primary")
-            out_f = gr.File(label="الملف الجاهز")
-            # محرك تحويل بسيط للعرض
-            go_btn.click(lambda x: None, inputs=editor, outputs=out_f)
+        return f"❌ خطأ في الخزنة الجديدة: {str(e)}"
+
+# واجهة بسيطة للتأكد من نجاح "البداية الجديدة"
+with gr.Blocks() as demo:
+    gr.Markdown("# 🔄 نظام إعادة التأسيس | AiOCR")
+    with gr.Row():
+        k = gr.Textbox(label="المفتاح")
+        t = gr.Textbox(label="نص التجربة الأولى")
+    btn = gr.Button("🚀 تدشين الخزنة الجديدة", variant="primary")
+    out = gr.Textbox(label="الحالة")
+    btn.click(fresh_start_upload, inputs=[k, t], outputs=out)
 
 demo.launch()