Update app.py

app.py

@@ -4,16 +4,19 @@ import os
 from utils import process_uploaded_file, export_to_docx
 from agents_config import create_agents
 
-def run_novel_studio(uploaded_file, manual_text, user_token):
-    if not user_token or not user_token.startswith("hf_"):
-        return "⚠️ يرجى إدخال Hugging Face Token صحيح (يبدأ بـ hf_).", None
+def run_novel_studio(uploaded_file, manual_text, oauth_token: gr.OAuthToken | None):
+    # جلب التوكن من تسجيل الدخول أو من متغيرات البيئة
+    user_token = oauth_token.token if oauth_token else os.getenv("HF_TOKEN")
+    
+    if not user_token:
+        return "⚠️ يرجى تسجيل الدخول عبر Hugging Face أولاً للوصول إلى النماذج.", None
 
     file_content = ""
     if uploaded_file is not None:
         try:
             file_content = process_uploaded_file(uploaded_file)
         except Exception as e:
-            return f"❌ خطأ في قراءة ملف docx: {str(e)}", None
+            return f"❌ خطأ في قراءة الملف: {str(e)}", None
 
     combined_context = f"{file_content}\n\n{manual_text}".strip()
     
@@ -21,6 +24,7 @@ def run_novel_studio(uploaded_file, manual_text, user_token):
         return "⚠️ المحتوى المقدم غير كافٍ. يرجى رفع ملف أو كتابة مسودة.", None
 
     try:
+        # إنشاء الوكلاء باستخدام التوكن المجلوب تلقائياً
         agents_dict = create_agents(user_token)
         
         assistant_agents = [
@@ -33,7 +37,7 @@ def run_novel_studio(uploaded_file, manual_text, user_token):
         groupchat = autogen.GroupChat(
             agents=assistant_agents,
             messages=[],
-            max_round=12,
+            max_round=15,
             speaker_selection_method="auto"
         )
         
@@ -42,11 +46,7 @@ def run_novel_studio(uploaded_file, manual_text, user_token):
             llm_config=agents_dict["editor"].llm_config
         )
 
-        init_message = f"""بصفتي رئيس التحرير، أضع بين أيديكم هذه المسودة. 
-        المطلوب: تحليلها، إكمال أحداث الفصل التالي، وتنسيقها أدبياً بشكل احترافي.
-        
-        المسودة:
-        {combined_context}"""
+        init_message = f"المسودة المطلوبة:\n{combined_context}"
         
         chat_result = agents_dict["editor"].initiate_chat(
             manager,
@@ -62,52 +62,29 @@ def run_novel_studio(uploaded_file, manual_text, user_token):
         return f"❌ فشل النظام في المعالجة: {str(e)}", None
 
 with gr.Blocks(theme=gr.themes.Soft(), title="Ling-1T Novel Studio") as demo:
-    gr.HTML("""
-        <div style="text-align: center;">
-            <h1>🖋️ Ling-1T Novel Production Studio</h1>
-            <p>نظام ذكاء اصطناعي متعدد الوكلاء لإنتاج وتنسيق الروايات العربية</p>
-        </div>
-    """)
-    
-    with gr.Row():
-        with gr.Column(scale=1):
-            gr.Markdown("### 🛠️ إعدادات المدخلات")
-            token_input = gr.Textbox(
-                label="Hugging Face Token", 
-                type="password", 
-                placeholder="hf_..."
-            )
-            file_upload = gr.File(
-                label="ارفع مسودة الرواية (.docx, .txt)", 
-                file_types=[".docx", ".txt"]
-            )
-            text_area = gr.Textbox(
-                label="تعليمات إضافية أو نص تكميلي", 
-                lines=6, 
-                placeholder="اكتب هنا إذا كنت تريد توجيه الوكلاء لمسار معين..."
-            )
-            submit_btn = gr.Button("🚀 إطلاق فريق العمل", variant="primary")
-            
-        with gr.Column(scale=2):
-            gr.Markdown("### 📖 المخطوطة النهائية")
-            output_markdown = gr.Markdown(
-                label="معاينة النص",
-                value="ستظهر الرواية هنا بعد اكتمال نقاش الوكلاء..."
-            )
-            download_btn = gr.File(label="تحميل ملف الوورد المنسق")
+    with gr.Sidebar():
+        gr.Markdown("# 🖋️ التحكم")
+        # إضافة زر تسجيل الدخول
+        gr.LoginButton()
+        
+        gr.Markdown("### 📂 ملقم المسودة")
+        file_upload = gr.File(label="ملف الرواية", file_types=[".docx", ".txt"])
+        text_area = gr.Textbox(label="تعليمات إضافية", lines=5)
+        submit_btn = gr.Button("🚀 إطلاق فريق العمل", variant="primary")
+
+    with gr.Column():
+        gr.HTML("<h1 style='text-align:center;'>Ling-1T Novel Production Studio</h1>")
+        output_markdown = gr.Markdown(value="بانتظار بدء العمل...")
+        download_btn = gr.File(label="تحميل المخطوطة النهائية (.docx)")
 
-    # التعديل ��نا: إضافة api_name=False لإصلاح خطأ Gradio 5.13
+    # ربط الأحداث مع إرسال oauth_token تلقائياً
     submit_btn.click(
         fn=run_novel_studio,
-        inputs=[file_upload, text_area, token_input],
+        inputs=[file_upload, text_area],
         outputs=[output_markdown, download_btn],
         api_name=False
     )
 
-    gr.Markdown("""
-    ---
-    **ملاحظة:** هذا النظام يستخدم 9 نماذج متخصصة تعمل بالتوازي. قد تستغرق العملية من 1-3 دقائق حسب طول النص.
-    """)
-
 if __name__ == "__main__":
     demo.launch()
+