Update app.py

app.py

@@ -1,61 +1,82 @@
+import gradio as gr
 import autogen
+import os
+from utils import process_uploaded_file, export_to_docx
+from agents_config import create_agents
 
-# النماذج التي أكدت فعاليتها - تم إزالة الملحقات اليدوية لضمان توافق الراوتر
-MODEL_MAP = {
-    "Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct",
-    "StyleGuardian": "openai/gpt-oss-120b", 
-    "Architect": "deepseek-ai/DeepSeek-V3.2-Exp",
-    "DraftWriter": "deepcogito/cogito-v2-preview-llama-405B",
-    "Humanizer": "deepcogito/cogito-671b-v2.1",
-    "ContinuityGuard": "meta-llama/Llama-3.3-70B-Instruct",
-    "Psychologist": "zai-org/GLM-4.7-FP8",
-    "Critic": "Qwen/Qwen3-235B-A22B-Thinking-2507",
-    "Editor": "inclusionAI/Ling-1T"
-}
-
-def get_llm_config(model_id, user_token):
-    return {
-        "config_list": [
-            {
-                "model": model_id,
-                "api_key": user_token,
-                "base_url": "https://router.huggingface.co/v1",
-                "api_type": "openai",
-            }
-        ],
-        "temperature": 0.7,
-        "cache_seed": None,
-    }
-
-def create_agents(user_token):
-    # إنشاء الوكلاء المساعدين
-    def mk_asst(name, msg, model_key):
-        return autogen.AssistantAgent(
-            name=name,
-            system_message=msg,
-            llm_config=get_llm_config(MODEL_MAP[model_key], user_token)
+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 "⚠️ يرجى تسجيل الدخول عبر زر (Sign in with 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"❌ خطأ في قراءة الملف: {str(e)}", None
+
+    combined_context = f"{file_content}\n\n{manual_text}".strip()
+    if len(combined_context) < 10:
+        return "⚠️ النص المقدم قصير جداً.", None
+
+    try:
+        agents_dict = create_agents(user_token)
+        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=assistant_agents,
+            messages=[],
+            max_round=15,
+            speaker_selection_method="auto"
         )
+        
+        manager = autogen.GroupChatManager(
+            groupchat=groupchat,
+            llm_config=agents_dict["editor"].llm_config
+        )
+
+        init_message = f"المسودة:\n{combined_context}"
+        chat_result = agents_dict["editor"].initiate_chat(manager, message=init_message)
+        
+        final_story_text = chat_result.chat_history[-1]['content']
+        output_file_path = export_to_docx(final_story_text)
+        
+        return final_story_text, output_file_path
+    except Exception as e:
+        return f"❌ خطأ تقني: {str(e)}", None
+
+# بناء الواجهة بشكل سليم
+with gr.Blocks(theme=gr.themes.Soft(), css="custom.css") as demo:
+    with gr.Sidebar():
+        gr.Markdown("# 🔑 تسجيل الدخول")
+        # الزر يجب أن يكون داخل Blocks حصراً
+        login_btn = gr.LoginButton()
+        
+        gr.Markdown("---")
+        gr.Markdown("### 📂 الملفات")
+        file_upload = gr.File(label="ارفع مسودة (.docx)", file_types=[".docx"])
+        text_area = gr.Textbox(label="أفكار إضافية", lines=8)
+        submit_btn = gr.Button("🚀 ابدأ العمل", variant="primary")
+            
+    with gr.Column():
+        gr.HTML("<h1 style='text-align:center;'>🖋️ Ling-1T Novel Studio</h1>")
+        output_markdown = gr.Markdown(value="سجل دخولك ثم اضغط 'ابدأ العمل'")
+        download_btn = gr.File(label="تحميل (.docx)")
 
-    agents = {
-        "analyst": mk_asst("Analyst", "أنت المحلل البنيوي والمفكك لأفكار الرواية.", "Analyst"),
-        "style_guardian": mk_asst("Style_Guardian", "أنت المسؤول عن النبرة والجمالية اللغوية.", "StyleGuardian"),
-        "architect": mk_asst("Architect", "أنت مهندس الحبكة وباني التسلسل الدرامي.", "Architect"),
-        "draft_writer": mk_asst("Draft_Writer", "أنت كاتب المسودة الأدبية بأسلوب روائي.", "DraftWriter"),
-        "humanizer": mk_asst("Humanizer", "أنت من يبث الروح والمشاعر في الحوارات والشخصيات.", "Humanizer"),
-        "continuity_guard": mk_asst("Continuity_Guard", "أنت مراقب المنطق ومنع التناقضات.", "ContinuityGuard"),
-        "psychologist": mk_asst("Psychologist", "أنت محلل الأعماق النفسية للشخصيات.", "Psychologist"),
-        "critic": mk_asst("Critic", "أنت الناقد الذي يراجع الجودة النهائية.", "Critic"),
-    }
-
-    # رئيس التحرير (Editor) - تم ضبطه للعمل التلقائي التام
-    agents["editor"] = autogen.UserProxyAgent(
-        name="Editor_In_Chief",
-        human_input_mode="NEVER", # منع التوقف لانتظار إدخال يدوي
-        max_consecutive_auto_reply=15,
-        code_execution_config=False,
-        llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
-        system_message="""أنت رئيس التحرير. مهمتك هي استلام الأفكار، الإشراف على نقاش الوكلاء، 
-        ثم تقديم النص الروائي النهائي في قالب منسق يحتوي على: العنوان، رقم الفصل، والمحتوى السردي."""
+    # الربط البرمجي (Gradio يمرر oauth_token تلقائياً)
+    submit_btn.click(
+        fn=run_novel_studio,
+        inputs=[file_upload, text_area],
+        outputs=[output_markdown, download_btn]
     )
 
-    return agents
+if __name__ == "__main__":
+    demo.launch()