Update agents_config.py

agents_config.py

@@ -1,110 +1,55 @@
-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, 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
+# القائمة المباشرة للنماذج
+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)
         )
 
-        # 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
+    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="أنت رئيس التحرير، اجمع النتائج في نص واحد."
     )
-
-if __name__ == "__main__":
-    demo.launch()
+    return agents