Update app.py

app.py

@@ -1,14 +1,13 @@
 import gradio as gr
-from transformers import AutoProcessor, LlavaForConditionalGeneration
+from transformers import AutoProcessor, Kosmos2Model
 import torch
 from PIL import Image
-import requests
 import io
 
-# 📦 تحميل النموذج والمُعالِج
-model_id = "llava-hf/llava-1.5-7b-hf"
-model = LlavaForConditionalGeneration.from_pretrained(model_id, device_map="auto")
+# 📦 تحميل المعالج والنموذج
+model_id = "microsoft/kosmos-2-patch14-224"
 processor = AutoProcessor.from_pretrained(model_id)
+model = Kosmos2Model.from_pretrained(model_id).to("cuda" if torch.cuda.is_available() else "cpu")
 
 # 💬 تعليمات الذكاء الاصطناعي لتكون خبير تداول
 SYSTEM_PROMPT = """
@@ -30,20 +29,26 @@ def analyze_chart(image):
         return "❌ لم يتم تحميل أي صورة."
 
     try:
-        # تحويل الصورة إلى تنسيق مناسب للنموذج
+        # فتح الصورة
         img = Image.open(io.BytesIO(image)).convert("RGB")
 
-        # إعداد prompt
-        prompt = f"<image>\nUSER: {SYSTEM_PROMPT}\nASSISTANT:"
+        # إنشاء prompt شامل
+        text = f"Question: {SYSTEM_PROMPT}\nAnswer:"
 
         # المعالجة
-        inputs = processor(prompt, images=img, return_tensors='pt').to(0, torch.float16)
+        inputs = processor(images=img, text=text, return_tensors="pt").to("cuda" if torch.cuda.is_available() else "cpu")
 
-        # التوليد
-        output = model.generate(**inputs, max_new_tokens=512)
-        response = processor.decode(output[0], skip_special_tokens=True)
+        # توليد التحليل
+        generated_ids = model.generate(
+            pixel_values=inputs["pixel_values"],
+            input_ids=inputs["input_ids"],
+            attention_mask=inputs["attention_mask"],
+            max_new_tokens=256
+        )
 
-        return response.split("ASSISTANT:")[-1].strip()
+        response = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
+
+        return response.strip()
 
     except Exception as e:
         return f"❌ حدث خطأ أثناء التحليل: {str(e)}"
@@ -53,10 +58,9 @@ interface = gr.Interface(
     fn=analyze_chart,
     inputs=gr.Image(type="bytes", label="تحميل مخطط التداول"),
     outputs=gr.Markdown(label="تحليل الذكاء الاصطناعي"),
-    title="🤖 منصة تحليل التداول الذكية (بدون API)",
+    title="🤖 منصة تحليل التداول الذكية (نسخة خفيفة)",
     description="استخدم الذكاء الاصطناعي المحلي لتحليل مخططات التداول وإعطاء تحليل احترافي.",
     theme="default",
-    examples=[["example_chart.png"]]
 )
 
 if __name__ == "__main__":