Update agents_config.py

agents_config.py

@@ -1,54 +1,110 @@
+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):
+    """
+    الدالة الرئيسية: تستخدم oauth_token لحل مشكلة الصلاحيات 403.
+    """
+    # 1. جلب التوكن من تسجيل الدخول أو متغيرات البيئة
+    user_token = oauth_token.token if oauth_token else os.getenv("HF_TOKEN")
+    
+    if not user_token:
+        return "⚠️ يرجى تسجيل الدخول أولاً عبر زر (Sign in with Hugging Face) لتمكين الوصول للنماذج.", None
+
+    # تعيين التوكن كمتغير بيئة لضمان عمل المكتبات التابعة
+    os.environ["HUGGINGFACEHUB_API_TOKEN"] = user_token
+
+    # 2. معالجة المدخلات
+    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:
+        # 3. إنشاء فريق الوكلاء باستخدام التوكن الموثق
+        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
         )
 
-    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"),
-    }
-
-    agents["editor"] = autogen.UserProxyAgent(
-        name="Editor_In_Chief",
-        human_input_mode="NEVER",
-        max_consecutive_auto_reply=10,
-        code_execution_config=False,
-        llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
-        system_message="أنت رئيس التحرير، اجمع النتائج في نص واحد."
+        # 4. بدء التنفيذ
+        init_message = f"إليك المادة الخام للرواية، ابدأوا المعالجة الجماعية فوراً:\n\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
+
+# إعداد واجهة Gradio
+with gr.Blocks(theme=gr.themes.Soft(), css="custom.css", title="Ling-1T Novel Studio") as demo:
+    with gr.Sidebar():
+        gr.Markdown("# 🔐 بوابة الوصول")
+        # حل مشكلة 403: طلب صلاحية Inference API
+        gr.LoginButton(scopes=["inference-api"])
+        
+        gr.Markdown("---")
+        gr.Markdown("### 📂 ملقم المسودة")
+        file_upload = gr.File(label="ارفع مسودة (.docx)", file_types=[".docx"])
+        text_area = gr.Textbox(
+            label="تعليمات إضافية", 
+            lines=6, 
+            placeholder="مثال: اجعل النهاية حزينة، أو ركز على البعد النفسي..."
+        )
+        submit_btn = gr.Button("🚀 إطلاق فريق العمل", variant="primary")
+            
+    with gr.Column():
+        gr.HTML("""
+            <div style='text-align:center;'>
+                <h1>🖋️ Ling-1T Novel Production Studio</h1>
+                <p>نظام وكلاء ذكاء اصطناعي لإنتاج الروايات العربية</p>
+            </div>
+        """)
+        output_markdown = gr.Markdown(value="**بانتظار تسجيل الدخول وبدء العملية...**")
+        download_btn = gr.File(label="تحميل المخطوطة النهائية (.docx)")
+
+    # ربط الأحداث مع تعطيل API Name لتجنب خطأ Schema في Gradio 5
+    submit_btn.click(
+        fn=run_novel_studio,
+        inputs=[file_upload, text_area],
+        outputs=[output_markdown, download_btn],
+        api_name=False
     )
-    return agents
+
+if __name__ == "__main__":
+    demo.launch()