Update genisi.js

genisi.js

@@ -8,24 +8,20 @@ import { fileURLToPath } from 'url';
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 const app = express();
 
-// إعدادات البيئة
 app.use(cors());
 app.use(express.json());
 app.use(express.static(__dirname));
 
-// تعريف الموديلات المستخدمة
 const MODELS = {
-    chat: "kimi-k2.5-fw",      // موديل الدردشة مع ميزة التفكير
-    code: "gemma-4-31b",     // موديل البرمجة السريع
-    image: "flux.1-dev"        // موديل توليد الصور
+    chat: "kimi-k2.5-fw",      // يدعم Reasoning
+    code: "gemma-4-31b",       // سريع، بدون reasoning
+    image: "flux.1-dev"
 };
 
-/**
- * دالة توليد الصور عبر NVIDIA API
- */
+// توليد الصور (بدون تغيير كبير)
 async function generateImage(prompt) {
     const NVIDIA_KEY = process.env.NVIDIA_API_KEY;
-    if (!NVIDIA_KEY) throw new Error("NVIDIA_API_KEY is missing!");
+    if (!NVIDIA_KEY) throw new Error("مفتاح NVIDIA API غير موجود!");
 
     const url = "https://ai.api.nvidia.com/v1/genai/black-forest-labs/flux.1-dev";
     const res = await fetch(url, {
@@ -35,64 +31,62 @@ async function generateImage(prompt) {
             "Content-Type": "application/json"
         },
         body: JSON.stringify({
-            "prompt": prompt,
-            "mode": "base", 
-            "cfg_scale": 3.5, 
-            "aspect_ratio": "1:1"
+            prompt,
+            mode: "base",
+            cfg_scale: 3.5,
+            aspect_ratio: "1:1"
         })
     });
     
     const result = await res.json();
-    if (!res.ok) throw new Error(result.detail || "NVIDIA Error");
+    if (!res.ok) throw new Error(result.detail || "خطأ في NVIDIA API");
     
-    // استخراج الصورة من هيكل رد NVIDIA
-    if (result && result.result && result.result.image) {
-        return result.result.image;
-    } else if (result.image) {
-        return result.image;
-    }
-    throw new Error("لم يتم العثور على بيانات الصورة في الرد");
+    return result.result?.image || result.image || null;
 }
 
 /**
- * المسار الرئيسي لمعالجة طلبات Genisi
+ * معالجة طلبات Genisi
+ * body: { action, message, reasoning? }
  */
 app.post('/api/genisi', async (req, res) => {
-    const { action, message } = req.body;
+    const { action, message, reasoning = false } = req.body;
     const POE_KEY = process.env.POE_API_KEY;
 
     if (!POE_KEY && action !== 'image') {
-        return res.status(500).json({ error: "POE_API_KEY missing" });
+        return res.status(500).json({ error: "مفتاح POE_API_KEY غير موجود" });
     }
 
     try {
+        // معالجة الصور (بدون stream)
         if (action === 'image') {
-            // معالجة الصور (بدون ستريم لأنها تعود كملف واحد)
             const imageData = await generateImage(message);
+            if (!imageData) throw new Error("لم يتم استلام بيانات الصورة");
             return res.json({ success: true, type: 'image', data: imageData });
         }
 
-        // إعداد البث المباشر (Streaming) للنصوص
+        // إعداد Streaming للنصوص
         res.setHeader('Content-Type', 'text/event-stream');
         res.setHeader('Cache-Control', 'no-cache');
         res.setHeader('Connection', 'keep-alive');
+        res.setHeader('X-Accel-Buffering', 'no'); // منع buffering في nginx
 
         const client = new OpenAI({
             apiKey: POE_KEY,
             baseURL: "https://api.poe.com/v1"
         });
 
-        // إعدادات الطلب
-        let requestOptions = {
+        // بناء الخيارات الأساسية
+        const requestOptions = {
             model: action === 'code' ? MODELS.code : MODELS.chat,
             messages: [{ role: "user", content: message }],
             stream: true,
+            // gemma يعمل بشكل افتراضي بدون reasoning
         };
 
-        // تفعيل ميزة التفكير (Reasoning) فقط لموديل Kimi
-        if (action === 'chat') {
+        // تفعيل Reasoning فقط لـ Kimi وعند الطلب الصريح
+        if (action === 'chat' && reasoning) {
             requestOptions.extra_body = {
-                "reasoning_effort": "high" 
+                reasoning_effort: "high" // أو "medium" حسب الرغبة
             };
         }
 
@@ -101,7 +95,6 @@ app.post('/api/genisi', async (req, res) => {
         for await (const chunk of stream) {
             const content = chunk.choices[0]?.delta?.content || "";
             if (content) {
-                // إرسال البيانات للمتصفح فوراً
                 res.write(`data: ${JSON.stringify({ content })}\n\n`);
             }
         }
@@ -110,7 +103,7 @@ app.post('/api/genisi', async (req, res) => {
         res.end();
 
     } catch (err) {
-        console.error("Internal Error:", err.message);
+        console.error("Server Error:", err);
         if (!res.headersSent) {
             res.status(500).json({ success: false, error: err.message });
         } else {
@@ -120,17 +113,15 @@ app.post('/api/genisi', async (req, res) => {
     }
 });
 
-// تشغيل الصفحة الرئيسية
 app.get('/', (req, res) => res.sendFile(path.join(__dirname, 'index.html')));
 
-const PORT = 7860;
+const PORT = process.env.PORT || 7860;
 app.listen(PORT, () => {
     console.log(`
-    🚀 Genisi Server is LIVE!
+    🚀 Genisi Server Running
     -------------------------
     Port: ${PORT}
-    Streaming: Enabled
-    Reasoning: Kimi (Active)
+    Reasoning: Kimi (On Demand) | Gemma (Fast)
     Developer: AnesNT
     -------------------------
     `);