Update app.py

app.py

@@ -4,38 +4,72 @@ from PIL import Image
 import io
 import numpy as np
 
-def generate_content(api_key, prompt, image_data):
+def generate_content(api_key, image_data, language):
+    genai.configure(api_key=api_key)
 
-    genai.configure(api_key=api_key)  
-
-    # Convert uploaded binary data into a PIL image.
-    # image = Image.open(io.BytesIO(image_data))
+    # Convert uploaded image data into a PIL image
+    if image_data is None:
+        return "Пожалуйста, загрузите изображение!"
     image = Image.fromarray(np.uint8(image_data)).convert('RGB')
 
+    # Base prompt
+    base_prompt = "Напиши большой и красивый, содержательный рекламный пост минимум на 1000 символов со смайликами и 25 тематических хэштегов с ключевыми словами разных вариантов, чтобы мои клиенты могли найти меня в поиске Instagram, Google и т.д. по ключевым словам. Пост пиши исключительно под товар, который на фото, без адресов и номеров телефона."
+
+    # Modify prompt based on selected language
+    if language == "Русский":
+        final_prompt = f"{base_prompt} Пиши на русском языке."
+    elif language == "Кыргызский":
+        final_prompt = f"{base_prompt} Пиши на кыргызском языке."
+    elif language == "Казахский":
+        final_prompt = f"{base_prompt} Пиши на казахском языке."
+    elif language == "Узбекский":
+        final_prompt = f"{base_prompt} Пиши на узбекском языке."
+    else:
+        final_prompt = base_prompt  # Default to Russian if no language selected
+
+    # Initialize the model
     model = genai.GenerativeModel('gemini-2.0-flash')
 
-    response = model.generate_content([prompt, image])
+    # Generate content with the image and prompt
+    response = model.generate_content([final_prompt, image])
     response.resolve()
     return response.text
 
+# Gradio interface
+with gr.Blocks(title="EVA - Генератор рекламных постов") as app:
+    gr.Markdown("# EVA - Загрузи изображение, выбери язык и получи красивый рекламный пост!")
 
+    # API Key input
+    api_key_input = gr.Textbox(
+        label="API Key",
+        type="password",
+        value="AIzaSyBtBdQiAQUeA4rKplFPuKd8zTP9r1LTVvA",
+        placeholder="Введите ваш API Key",
+        lines=1
+    )
 
-with gr.Blocks() as app:
-    gr.Markdown("# EVA - загрузи изображение, нажми пуск и получишь красивый рекламный пост ")
-    gr.Markdown("https://ai.google.dev/")
-    api_key_input = gr.Textbox(label="API Key", type="password", value="AIzaSyBtBdQiAQUeA4rKplFPuKd8zTP9r1LTVvA", placeholder="Enter your API Key", lines=1)
+    # Language selection
+    language_input = gr.Radio(
+        choices=["Русский", "Кыргызский", "Казахский", "Узбекский"],
+        label="Выберите язык поста",
+        value="Русский"  # Default value
+    )
 
-    # image_input = gr.File(label="Upload an image", type="binary")
-    image_input = gr.Image(label="Загрузить изображение ")
-    prompt_input = gr.Textbox(label="Функция", placeholder="Enter your Prompt", type="password",value="Напиши большой и красивый, содержательный рекламный пост миниммум на 1000 символов со смайликами и 25 тематических хэштегов с ключевыми словами разных вариантов , чтобы мои клиенты могли найти меня в поиске instagram, Google итд по ключевым словам. все исключительно на Русском языке , без адресов и номеров телефона.пост пиши исключительно под товар который на фото    " ,lines=1)
+    # Image input
+    image_input = gr.Image(label="Загрузить изображение")
+
+    # Generate button
     generate_button = gr.Button("Пуск")
-    output = gr.Markdown()
-    
+
+    # Output
+    output = gr.Markdown(label="Результат")
+
+    # Button click event
     generate_button.click(
-        generate_content, 
-        inputs=[api_key_input, prompt_input, image_input],
+        fn=generate_content,
+        inputs=[api_key_input, image_input, language_input],
         outputs=output
     )
 
-app.launch()
-
+# Launch the app
+app.launch()