Update app.py

app.py

@@ -4,132 +4,123 @@ import os
 from utils import process_uploaded_file, export_to_docx
 from agents_config import create_agents
 
-def start_novel_engine(file_obj, extra_text, user_token):
-    if not user_token:
-        return "⚠️ يرجى إدخال التوكن الخاص بك.", None
-    
-    # 1. استخراج النص من الملف
-    context = ""
-    if file_obj is not None:
+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
+
+    # 1. معالجة المدخلات (الملف المرفوع + النص اليدوي)
+    file_content = ""
+    if uploaded_file is not None:
         try:
-            context = process_uploaded_file(file_obj)
+            file_content = process_uploaded_file(uploaded_file)
         except Exception as e:
-            return f"❌ خطأ في الملف: {str(e)}", None
+            return f"❌ خطأ في قراءة ملف docx: {str(e)}", None
+
+    # دمج المحتوى مع الحفاظ على الأسطر
+    combined_context = f"{file_content}\n\n{manual_text}".strip()
     
-    full_prompt = f"{context}\n\n{extra_text}".strip()
-    if len(full_prompt) < 10:
-        return "⚠️ النص قصير جدًا للبدء.", None
+    if len(combined_context) < 10:
+        return "⚠️ المحتوى المقدم غير كافٍ. يرجى رفع ملف أو كتابة مسودة.", None
 
     try:
-        # 2. تشغيل الوكلاء
-        agents = create_agents(user_token)
+        # 2. إنشاء فريق الوكلاء الـ 9
+        agents_dict = create_agents(user_token)
+        
+        # 3. إعداد غرفة النقاش (Group Chat)
+        # الوكلاء المساعدون (بدون المدير ورئيس التحرير)
+        assistant_agents = [
+            agents_dict["analyst"], agents_dict["style_guardian"],
+            agents_dict["architect"], agents_dict["draft_writer"],
+            agents_dict["humanizer"], agents_dict["continuity_guard"],
+            agents_dict["psychologist"], agents_dict["critic"]
+        ]
+        
         groupchat = autogen.GroupChat(
-            agents=[v for k, v in agents.items() if k != "editor"],
+            agents=assistant_agents,
             messages=[],
-            max_round=10
+            max_round=12,
+            speaker_selection_method="auto"
         )
-        manager = autogen.GroupChatManager(groupchat=groupchat, llm_config=agents["editor"].llm_config)
         
-        # 3. التواصل مع Ling-1T
-        chat_result = agents["editor"].initiate_chat(
-            manager, 
-            message=f"قم بصياغة الفصل التالي وتنسيقه أدبيًا:\n\n{full_prompt}"
+        manager = autogen.GroupChatManager(
+            groupchat=groupchat,
+            llm_config=agents_dict["editor"].llm_config
         )
+
+        # 4. إطلاق المهمة بقيادة Ling-1T
+        init_message = f"""بصفتي رئيس التحرير، أضع بين أيديكم هذه المسودة. 
+        المطلوب: تحليلها، إكمال أحداث الفصل التالي، وتنسيقها أدبياً بشكل احترافي.
         
-        final_text = chat_result.chat_history[-1]['content']
-        docx_path = export_to_docx(final_text)
+        المسودة:
+        {combined_context}"""
         
-        return final_text, docx_path
-    except Exception as e:
-        return f"❌ خطأ أثناء الإنتاج: {str(e)}", None
-
-# واجهة مستقرة (بدون SSR وبدون مسببات الخطأ)
-with gr.Blocks(analytics_enabled=False) as demo:
-    gr.Markdown("# 🖋️ Ling-1T Novel Studio")
-    
-    with gr.Row():
-        with gr.Column():
-            hf_token = gr.Textbox(label="Hugging Face Token", type="password")
-            # إزالة أي خصائص إضافية من الملف لضمان الاستقرار
-            file_input = gr.File(label="Upload DOCX/TXT")
-            instruction = gr.Textbox(label="Special Instructions", lines=3)
-            run_btn = gr.Button("🚀 Start Writing", variant="primary")
-            
-        with gr.Column():
-            output_display = gr.Textbox(label="Manuscript Preview", lines=12)
-            download_link = gr.File(label="Download Final DOCX")
-
-    # القضاء على الخطأ عبر منع الـ API تمامًا لهذا الحدث
-    run_btn.click(
-        fn=start_novel_engine,
-        inputs=[file_input, instruction, hf_token],
-        outputs=[output_display, download_link],
-        api_name=False  # هذا السطر يمنع Gradio من فحص المكونات برمجياً
-    )
-
-if __name__ == "__main__":
-    # تشغيل مباشر وبسيط
-    demo.launch(show_api=False)
-    utils import process_uploaded_file, export_to_docx
-from agents_config import create_agents
-
-# الدالة الأساسية (تأكد أن الأسماء تطابق ما في utils.py)
-def process_all(file_obj, manual_text, token):
-    if not token:
-        return "❌ يرجى إدخال Token", None
-    
-    # قراءة الملف
-    context = ""
-    if file_obj is not None:
-        context = process_uploaded_file(file_obj)
-    
-    context += "\n" + manual_text
-    
-    if len(context.strip()) < 10:
-        return "⚠️ النص قصير جداً", None
-
-    try:
-        agents = create_agents(token)
-        # إعداد الحوار
-        groupchat = autogen.GroupChat(
-            agents=[v for k, v in agents.items() if k != "editor"],
-            messages=[],
-            max_round=8
+        chat_result = agents_dict["editor"].initiate_chat(
+            manager,
+            message=init_message
         )
-        manager = autogen.GroupChatManager(groupchat=groupchat, llm_config=agents["editor"].llm_config)
         
-        # التنفيذ
-        chat_result = agents["editor"].initiate_chat(manager, message=f"أكمل ونسق النص: {context}")
+        # 5. استخراج النص النهائي من Ling-1T وتوليد ملف DOCX
+        # نبحث عن آخر رسالة تحتوي على النص المدمج
+        final_story_text = chat_result.chat_history[-1]['content']
         
-        final_text = chat_result.chat_history[-1]['content']
-        docx_file = export_to_docx(final_text)
+        # توليد الملف باستخدام دالة utils المحدثة
+        output_file_path = export_to_docx(final_story_text)
         
-        return final_text, docx_file
+        return final_story_text, output_file_path
+
     except Exception as e:
-        return f"❌ خطأ تقني: {str(e)}", None
+        return f"❌ فشل النظام في المعالجة: {str(e)}", None
 
-# واجهة مستقرة جداً (Minimalist Approach)
-with gr.Blocks(ssr=False) as demo: # تعطيل SSR هو مفتاح الحل هنا
-    gr.Markdown("# 🖋️ Ling-1T Novel Production")
+# تصميم واجهة المستخدم (تنسيق احترافي وهادئ)
+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():
-            token_input = gr.Textbox(label="HF Token", type="password")
-            # تبسيط الـ File component لأقصى حد
-            file_input = gr.File(label="Upload DOCX/TXT") 
-            text_input = gr.Textbox(label="Additional Text", lines=3)
-            run_btn = gr.Button("🚀 Start")
+        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():
-            output_text = gr.Textbox(label="Manuscript", lines=10)
-            output_file = gr.File(label="Download DOCX")
+        with gr.Column(scale=2):
+            gr.Markdown("### 📖 المخطوطة النهائية")
+            output_markdown = gr.Markdown(
+                label="معاينة النص",
+                value="ستظهر الرواية هنا بعد اكتمال نقاش الوكلاء..."
+            )
+            download_btn = gr.File(label="تحميل ملف الوورد المنسق")
 
-    run_btn.click(
-        fn=process_all,
-        inputs=[file_input, text_input, token_input],
-        outputs=[output_text, output_file],
-        api_name=False # منع توليد API لهذا الزر تحديداً
+    # ربط الأحداث
+    submit_btn.click(
+        fn=run_novel_studio,
+        inputs=[file_upload, text_area, token_input],
+        outputs=[output_markdown, download_btn]
     )
 
+    gr.Markdown("""
+    ---
+    **ملاحظة:** هذا النظام يستخدم 9 نماذج متخصصة تعمل بالتوازي. قد تستغرق العملية من 1-3 دقائق حسب طول النص.
+    """)
+
 if __name__ == "__main__":
-    demo.launch(show_api=False) # إغلاق الـ API تماماً
+    demo.launch()