Update app.py

app.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 import os
 import sys
 import base64
@@ -169,8 +168,8 @@ class StoryGenerator:
         self.site_url = os.getenv("SITE_URL", "https://rawi-kids-stories.com")
         self.site_name = os.getenv("SITE_NAME", "Rawi Kids Stories")
         
-        # OpenRouter model for vision capabilities - Updated to use Llama Vision
-        self.vision_model = "meta-llama/llama-3.2-11b-vision-instruct:free"
+        # OpenRouter model for vision capabilities
+        self.vision_model = "openai/gpt-4.1"
         
         # Story template for kids ages 6-10
         self.template = "اكتب قصة بسيطة وممتعة لطفل يتراوح عمره بين 6-10 سنوات عن: {}. حافظ على أن تكون القصة حوالي 300-400 كلمة بلغة بسيطة ورسالة إيجابية. اكتب القصة باللغة العربية الفصحى المناسبة للأطفال. تأكد من أن القصة تحترم الثقافة والأخلاق العربية والإسلامية. في نهاية القصة، اختم بعبارة واضحة تبدأ بـ 'ومن هذه القصة نتعلم أن...' توضح الدرس الإيجابي المستفاد من القصة."
@@ -200,7 +199,7 @@ class StoryGenerator:
         }
     
     def get_image_description(self, image_bytes):
-        """Get a detailed description of the image using OpenRouter's Llama Vision model"""
+        """Get a detailed description of the image using OpenRouter's GPT-4.1"""
         if not self.openrouter_api_key:
             raise ValueError("OpenRouter API key not set")
             
@@ -211,7 +210,7 @@ class StoryGenerator:
         prompt = "صف هذه الصورة بالتفصيل. ركز على العناصر الرئيسية والأعمال والإعدادات وأي عناصر ملحوظة. قدم وصفًا واضحًا وشاملًا يمكن استخدامه كسياق لكتابة قصة للأطفال. الرجاء الرد باللغة العربية."
         
         try:
-            # Call the OpenRouter API with Llama Vision model
+            # Call the OpenRouter API with GPT-4.1 model
             headers = {
                 "Content-Type": "application/json",
                 "Authorization": f"Bearer {self.openrouter_api_key}",
@@ -332,7 +331,7 @@ class StoryGenerator:
             image.save(buffered, format="JPEG", quality=85)
             img_bytes = buffered.getvalue()
             
-            # Step 1: Get image description using OpenRouter/Llama Vision
+            # Step 1: Get image description using OpenRouter/GPT-4.1
             description = self.get_image_description(img_bytes)
             
             # Step 2: Generate story from description using DeepSeek (with random theme)
@@ -625,4 +624,4 @@ with gr.Blocks(title="Rawi Kids Story Generator") as demo:
 
 # Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()