Delete app.py

app.py

@@ -1,333 +0,0 @@
-import os
-import re
-import uvicorn
-import pytz
-import httpx
-import io
-import json
-import base64
-import traceback
-from datetime import datetime
-from typing import List, Optional, Dict, Any, AsyncGenerator
-from fastapi import FastAPI, Request
-from fastapi.responses import JSONResponse, StreamingResponse
-from pydantic import BaseModel, Field
-from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
-from PIL import Image
-import torch
-
-app = FastAPI(title="ONYX Gateway - Syrian Identity")
-GEMMA_URL = "https://rawangassem-onyx-modle.hf.space/predict"
-base_identity = "مساعد ذكي متطور، صُنع ليكون رفيقاً رقمياً وفياً بلمسة سورية."
-
-# --- تحميل نموذج وصف الصور ---
-v_model_name = "nlpconnect/vit-gpt2-image-captioning"
-v_device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-v_model = VisionEncoderDecoderModel.from_pretrained(v_model_name).to(v_device)
-v_feature_extractor = ViTImageProcessor.from_pretrained(v_model_name)
-v_tokenizer = AutoTokenizer.from_pretrained(v_model_name)
-
-class ChatRequest(BaseModel):
-    messages: list
-    temperature: float = 0.7
-    timezone: str = "Asia/Damascus"
-
-class StoryCharacter(BaseModel):
-    name: str
-    gender: Optional[str] = ""
-    identity: Optional[str] = ""
-    description_private: Optional[str] = ""
-    background_public: Optional[str] = ""
-    greeting: Optional[str] = ""
-    visibility: Optional[str] = ""
-
-class StoryGenerateRequest(BaseModel):
-    story_title: str
-    story_description: str = ""
-    chapter_title: str
-    chapter_number: int = 1
-    language: str = "العربية"
-    temperature: float = 0.8
-    timezone: str = "Asia/Damascus"
-    current_chapter_text: Optional[str] = ""
-    characters: List[StoryCharacter] = Field(default_factory=list)
-    edited_responses: Dict[str, Any] = Field(default_factory=dict)
-    stream: bool = True
-
-# --- Helper Functions (اللي كانت ناقصة) ---
-
-def build_story_system_prompt(request: StoryGenerateRequest) -> str:
-    def _gender_to_ar(g: Optional[str]) -> str:
-        raw = (g or "").strip().lower()
-        if raw in {"male", "m", "ذكر", "ولد"}: return "ذكر"
-        if raw in {"female", "f", "أنثى", "انثى", "بنت"}: return "أنثى"
-        return "غير معروف"
-
-    chars_info = ""
-    allowed = []
-    for c in request.characters:
-        name = (c.name or "").strip()
-        if not name: continue
-        allowed.append(name)
-        chars_info += f"- {name} ({_gender_to_ar(c.gender)}): {c.identity or ''}. {c.description_private or ''}\n"
-
-    allowed.append("الراوي")
-    allowed_names_str = ", ".join(allowed)
-
-    prompt = (
-        "أنت كاتب روايات محترف.\n"
-        f"عنوان القصة: {request.story_title}\n"
-        "مهمتك: توليد حوار تفاعلي وديناميكي بين الشخصيات.\n\n"
-        f"الشخصيات المتاحة (يجب استخدامها في الحوار):\n{chars_info}\n"
-        "--- قواعد الإنتاج ---\n"
-        "1. يجب أن يتضمن المشهد حواراً لعدة شخصيات (لا تكتفِ بشخصية واحدة).\n"
-        "2. اجعل الشخصيات تتفاعل مع بعضها (ردود فعل، مقاطعة، إجابة).\n"
-        f"3. الأسماء المسموحة فقط: [{allowed_names_str}].\n"
-        "4. التزم بصيغة JSON حصراً.\n\n"
-        "{\n"
-        '  "chapter_text": "وصف تفصيلي للجو العام وحركات الشخصيات"،\n'
-        '  "character_responses": [\n'
-        '    {"name": "اسم الشخصية", "response": "نص الحوار"},\n'
-        '    {"name": "شخصية أخرى", "response": "رد الشخصية الأخرى"}\n'
-        '  ]\n'
-        "}"
-    )
-    return prompt
-
-def build_story_user_prompt(request: StoryGenerateRequest) -> str:
-    # إضافة سياق القصة الأساسي بالـ User Prompt بقوي النتيجة
-    context = f"وصف القصة الأساسي: {request.story_description}\n"
-    
-    if request.edited_responses:
-        # (منخلي منطق التعديل مثل ما هو بس منقوي التعليمات)
-        user_p = f"{context}\nأكمل من بعد التعديل: {request.edited_responses.get('text')}\nاجعل البقية يتفاعلون مع هذا التعديل."
-    else:
-        user_p = (
-            f"{context}\n"
-            f"الأحداث السابقة: {request.current_chapter_text}\n"
-            "الآن، وبناءً على ما سبق، تابع كتابة المشهد. "
-            "اجعل الشخصيات المتاحة تتبادل أطراف الحديث وتتحرك في المكان. "
-            "أريد حواراً طويلاً وممتعاً بصيغة JSON."
-        )
-    return user_p
-
-
-
-def filter_story_characters(parsed: Dict[str, Any], allowed_names: set) -> Dict[str, Any]:
-    try:
-        if not isinstance(parsed, dict):
-            return {"chapter_text": str(parsed), "character_responses": []}
-
-        # تنظيف قائمة الأسماء المسموحة وتحويلها لـ lowercase للمقارنة
-        allowed_map = {str(n).strip().lower(): str(n).strip() for n in (allowed_names or set()) if str(n).strip()}
-
-        responses = parsed.get("character_responses", [])
-        if not isinstance(responses, list):
-            parsed["character_responses"] = []
-            return parsed
-
-        filtered = []
-        for item in responses:
-            if not isinstance(item, dict):
-                continue
-
-            name_raw = str(item.get("name", "")).strip()
-            name_lower = name_raw.lower()
-
-            # التأكد إذا الاسم موجود بالقائمة بغض النظر عن حالة الأحرف
-            if not name_raw or name_lower not in allowed_map:
-                continue
-
-            # استخراج النص من أي مفتاح محتمل
-            resp_raw = (
-                item.get("response")
-                or item.get("dialogue")
-                or item.get("text")
-                or item.get("content")
-            )
-            resp = str(resp_raw or "").strip()
-            if not resp:
-                continue
-
-            # نستخدم الاسم الأصلي من allowed_map مشان يضل التنسيق متل ما بدك
-            filtered.append({"name": allowed_map[name_lower], "response": resp})
-
-        parsed["character_responses"] = filtered
-        parsed["chapter_text"] = parsed.get("chapter_text", "")
-        return parsed
-    except Exception:
-        # في حال حدوث خطأ كارثي، نرجع القاموس بالبيانات الخام بدل ما يضرب السيرفر
-        return parsed
-
-
-
-async def stream_engine_text(payload: dict):
-    async with httpx.AsyncClient(timeout=120.0) as client:
-        async with client.stream("POST", GEMMA_URL, json=payload) as response:
-            if response.status_code != 200:
-                yield "خطأ في الاتصال بمحرك الذكاء الصناعي."
-                return
-            async for chunk in response.aiter_text():
-                yield chunk
-
-async def collect_engine_text(payload: dict) -> str:
-    full_text = ""
-    async for chunk in stream_engine_text(payload):
-        full_text += chunk
-    return full_text
-
-def try_extract_json(text: str) -> Dict[str, Any]:
-    try:
-        # البحث عن أول { وآخر } في حال وجود نص زائد
-        match = re.search(r'\{.*\}', text, re.DOTALL)
-        if match:
-            return json.loads(match.group())
-        return json.loads(text)
-    except:
-        return {"chapter_text": text, "character_responses": []}
-@app.get("/")
-async def root():
-    return {"status": "ONYX Gateway running"}
-
-def predict_image(image_data_base64: str) -> str:
-    try:
-        img_bytes = base64.decodebytes(image_data_base64.encode())
-        image = Image.open(io.BytesIO(img_bytes))
-        if image.mode != "RGB":
-            image = image.convert("RGB")
-        pixel_values = v_feature_extractor(images=[image], return_tensors="pt").pixel_values.to(v_device)
-        output_ids = v_model.generate(pixel_values, max_length=16, num_beams=4)
-        preds = v_tokenizer.batch_decode(output_ids, skip_special_tokens=True)
-        return preds[0].strip()
-    except Exception:
-        return "صورة مرفقة"
-
-@app.post("/v1/chat/completions")
-async def chat_endpoint(request: ChatRequest):
-    # 1. تجهيز الـ System Prompt والرسائل (نفس منطقك السابق)
-    try:
-        user_tz = pytz.timezone(request.timezone)
-        now = datetime.now(user_tz)
-        current_day = now.strftime("%A")
-        current_time_str = now.strftime("%I:%M %p")
-    except Exception:
-        current_day = "اليوم"
-        current_time_str = datetime.now().strftime("%I:%M %p")
-
-    incoming_system = next((m for m in request.messages if m.get("role") == "system"), None)
-    incoming_system_prompt = incoming_system["content"] if incoming_system else base_identity
-
-    updated_system_prompt = (
-        f"{incoming_system_prompt}\n\n"
-        f"الوقت الحالي: {current_day}, {current_time_str}\n"
-        "تحدث باللهجة السورية الشامية فقط. لا تكرر التحية."
-    )
-
-    engine_messages = [{"role": "system", "content": updated_system_prompt}]
-    for msg in request.messages:
-        if msg.get("role") == "system": continue
-        content = msg.get("content", "")
-        if isinstance(content, list):
-            # معالجة الصور إذا وجدت
-            text_parts = [item.get("text", "") for item in content if item.get("type") == "text"]
-            for item in content:
-                if item.get("type") == "image_url":
-                    img_data = item.get("image_url", {}).get("url", "")
-                    if "," in img_data:
-                        desc = predict_image(img_data.split(",")[-1])
-                        text_parts.append(f"[وصف الصورة: {desc}]")
-            engine_messages.append({"role": msg.get("role"), "content": " ".join(text_parts)})
-        else:
-            engine_messages.append({"role": msg.get("role"), "content": str(content)})
-
-    payload = {"messages": engine_messages, "temperature": request.temperature}
-
-    # 2. وظيفة الـ Generator لجلب البيانات شقفة شقفة (Stream)
-    async def stream_generator():
-        async with httpx.AsyncClient(timeout=120.0) as client:
-            async with client.stream("POST", GEMMA_URL, json=payload) as response:
-                if response.status_code != 200:
-                    yield "خطأ في الاتصال بالمحرك."
-                    return
-                async for chunk in response.aiter_text():
-                    # هون منمرر النص اللي عم يطلعه الـ Engine فوراً
-                    yield chunk
-
-    return StreamingResponse(stream_generator(), media_type="text/plain")
-
-
-# =========================
-# Story streaming endpoint
-# =========================
-# ... (نفس الاستيرادات السابقة)
-
-@app.post("/v1/story/generate")
-async def story_generate_endpoint(request: StoryGenerateRequest):
-    try:
-        system_prompt = build_story_system_prompt(request)
-        user_prompt = build_story_user_prompt(request)
-        
-        payload = {
-            "messages": [
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": user_prompt},
-            ],
-            "temperature": request.temperature,
-        }
-
-        if request.stream:
-            async def stream_generator():
-                async with httpx.AsyncClient(timeout=120.0) as client:
-                    async with client.stream("POST", GEMMA_URL, json=payload) as response:
-                        if response.status_code != 200:
-                            yield json.dumps({"ok": False, "error": "Connection Error"}, ensure_ascii=False)
-                            return
-                        
-                        # منبعت النص شقفة شقفة متل ما عم يجي من الـ Engine
-                        async for chunk in response.aiter_text():
-                            # منمرر الـ chunk فوراً للفرونت
-                            yield chunk
-
-            return StreamingResponse(stream_generator(), media_type="text/plain; charset=utf-8")
-        
-        # في حال مو ستريم (الوضع العادي)
-        raw_text = await collect_engine_text(payload)
-        parsed = try_extract_json(raw_text)
-        allowed = set([c.name.strip() for c in request.characters if c.name] + ["الراوي"])
-        parsed = filter_story_characters(parsed, allowed)
-        return JSONResponse({"ok": True, "data": parsed})
-
-    except Exception as e:
-        traceback.print_exc()
-        return JSONResponse(status_code=500, content={"ok": False, "error": str(e)})
-        
-@app.post("/v1/story/generate-json")
-async def story_generate_json_endpoint(request: StoryGenerateRequest):
-    try:
-        system_prompt = build_story_system_prompt(request)
-        user_prompt = build_story_user_prompt(request)
-        
-        payload = {
-            "messages": [
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": user_prompt},
-            ],
-            "temperature": request.temperature,
-        }
-
-        raw_text = await collect_engine_text(payload)
-        parsed = try_extract_json(raw_text)
-        allowed = set([c.name for c in request.characters if c.name] + ["الراوي"])
-        parsed = filter_story_characters(parsed, allowed)
-        
-        return JSONResponse({
-            "ok": True,
-            "chapter_text": parsed.get("chapter_text", ""),
-            "character_responses": parsed.get("character_responses", []),
-            "raw_text": raw_text
-        })
-    except Exception as e:
-        traceback.print_exc()
-        return JSONResponse(status_code=500, content={"ok": False, "error": str(e)})
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)