Update main.py

main.py

@@ -69,7 +69,7 @@ async def chat_endpoint(request: ChatRequest):
                 # ترجمة الطلب لـ English لضمان جودة الصور
                 client = get_text_client()
                 translation_response = client.models.generate_content(
-                    model="gemini-1.5-flash", 
+                    model="gemma-4-31b-it", 
                     contents=f"Translate this prompt to English for an image generator. Only return the English prompt: {request.message}"
                 )
                 final_prompt = translation_response.text.strip()
@@ -183,7 +183,7 @@ async def chat_endpoint(request: ChatRequest):
 AnesNT مبادرة جزائرية من ولاية باتنة صاحبها انس كامش باللاتينية Anes Kameche وهو حاليا العضو الوحيد في هذي المبادرة وهي مبادرة تكنولوجية
 كن مفيدا وصديقا للمستخدم تحدث باللغة التي تحدث بها المستخدم"""
 
-            chosen_model = "gemini-1.5-pro-latest" if has_images else "gemma-4-31b-it"
+            chosen_model = "gemma-4-31b-it" if has_images else "gemma-4-31b-it"
 
             # تم دمج thinking_config هنا
             config = types.GenerateContentConfig(