Update app.py

app.py

@@ -1,132 +1,65 @@
 import gradio as gr
 import autogen
-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
-
-    # 1. معالجة المدخلات (الملف المرفوع + النص اليدوي)
-    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
-
-    # دمج المحتوى مع الحفاظ على الأسطر
-    combined_context = f"{file_content}\n\n{manual_text}".strip()
+# الدالة الأساسية (تأكد أن الأسماء تطابق ما في 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(combined_context) < 10:
-        return "⚠️ المحتوى المقدم غير كافٍ. يرجى رفع ملف أو كتابة مسودة.", None
+    if len(context.strip()) < 10:
+        return "⚠️ النص قصير جداً", None
 
     try:
-        # 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"]
-        ]
-        
+        agents = create_agents(token)
+        # إعداد الحوار
         groupchat = autogen.GroupChat(
-            agents=assistant_agents,
+            agents=[v for k, v in agents.items() if k != "editor"],
             messages=[],
-            max_round=12,
-            speaker_selection_method="auto"
+            max_round=8
         )
+        manager = autogen.GroupChatManager(groupchat=groupchat, llm_config=agents["editor"].llm_config)
         
-        manager = autogen.GroupChatManager(
-            groupchat=groupchat,
-            llm_config=agents_dict["editor"].llm_config
-        )
-
-        # 4. إطلاق المهمة بقيادة Ling-1T
-        init_message = f"""بصفتي رئيس التحرير، أضع بين أيديكم هذه المسودة. 
-        المطلوب: تحليلها، إكمال أحداث الفصل التالي، وتنسيقها أدبياً بشكل احترافي.
+        # التنفيذ
+        chat_result = agents["editor"].initiate_chat(manager, message=f"أكمل ونسق النص: {context}")
         
-        المسودة:
-        {combined_context}"""
+        final_text = chat_result.chat_history[-1]['content']
+        docx_file = export_to_docx(final_text)
         
-        chat_result = agents_dict["editor"].initiate_chat(
-            manager,
-            message=init_message
-        )
-        
-        # 5. استخراج النص النهائي من Ling-1T وتوليد ملف DOCX
-        # نبحث عن آخر رسالة تحتوي على النص المدمج
-        final_story_text = chat_result.chat_history[-1]['content']
-        
-        # توليد الملف باستخدام دالة utils المحدثة
-        output_file_path = export_to_docx(final_story_text)
-        
-        return final_story_text, output_file_path
-
+        return final_text, docx_file
     except Exception as e:
-        return f"❌ فشل النظام في المعالجة: {str(e)}", None
+        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>
-    """)
+# واجهة مستقرة جداً (Minimalist Approach)
+with gr.Blocks(ssr=False) as demo: # تعطيل SSR هو مفتاح الحل هنا
+    gr.Markdown("# 🖋️ Ling-1T Novel Production")
     
     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():
+            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=2):
-            gr.Markdown("### 📖 المخطوطة النهائية")
-            output_markdown = gr.Markdown(
-                label="معاينة النص",
-                value="ستظهر الرواية هنا بعد اكتمال نقاش الوكلاء..."
-            )
-            download_btn = gr.File(label="تحميل ملف الوورد المنسق")
+        with gr.Column():
+            output_text = gr.Textbox(label="Manuscript", lines=10)
+            output_file = gr.File(label="Download DOCX")
 
-    # ربط الأحداث
-    submit_btn.click(
-        fn=run_novel_studio,
-        inputs=[file_upload, text_area, token_input],
-        outputs=[output_markdown, download_btn]
+    run_btn.click(
+        fn=process_all,
+        inputs=[file_input, text_input, token_input],
+        outputs=[output_text, output_file],
+        api_name=False # منع توليد API لهذا الزر تحديداً
     )
 
-    gr.Markdown("""
-    ---
-    **ملاحظة:** هذا النظام يستخدم 9 نماذج متخصصة تعمل بالتوازي. قد تستغرق العملية من 1-3 دقائق حسب طول النص.
-    """)
-
 if __name__ == "__main__":
-    # إضافة show_api=False تمنع Gradio من محاولة فحص المكونات وتوليد الـ API
-    # وهذا هو الحل الجذري لمشكلة TypeError: argument of type 'bool'
-    demo.launch(
-        server_name="0.0.0.0", 
-        server_port=7860, 
-        show_api=False
-    )
+    demo.launch(show_api=False) # إغلاق الـ API تماماً