Create app.py

app.py

@@ -0,0 +1,118 @@
+import gradio as gr
+import requests
+import os
+import base64
+from PIL import Image
+import io
+import random
+
+# Функция для кодирования изображения в base64
+def encode_image_to_base64(image):
+    buffered = io.BytesIO()
+    image.save(buffered, format="JPEG")
+    img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    return img_str
+
+emojis = ['😊', '🤗', '🥺', '😅', '🤭', '😔', '✨', '😜', '🙏']
+
+# Функция для отправки запроса в OpenAI с изображением и получения ответа
+def ask_openai_with_image(instruction, image):
+    if not instruction and not image:
+        emj = random.choice(emojis)
+        raise gr.Error(f"{emj} Заполни пожалуйста хотя бы 1 поле")
+        return None
+    if image:
+        # Кодируем загруженное изображение в base64
+        base64_image = encode_image_to_base64(image)
+
+        # Создаем данные для запроса с закодированным изображением
+        payload = {
+            "model": "gpt-4-vision-preview",
+            "messages": [
+                {
+                    "role": "user",
+                    "content": [
+                        {
+                            "type": "text",
+                            "text": start,
+                        },
+                        {
+                            "type": "text",
+                            "text": instruction,
+                        },
+                        {
+                            "type": "image_url",
+                            "image_url": {
+                                "url": f"data:image/jpeg;base64,{base64_image}",
+                                "detail": "high",
+                            },
+                        },
+                    ],
+                }
+            ],
+            "max_tokens": 4095,
+        }
+    else:
+        # Создаем данные для запроса без изображения
+        payload = {
+            "model": "gpt-4-vision-preview",
+            "messages": [
+                {
+                    "role": "user",
+                    "content": start,
+                },
+                {
+                    "role": "user",
+                    "content": instruction,
+                }
+            ],
+            "max_tokens": 4095,
+        }    
+
+    # API ключ для OpenAI
+    api_key = os.getenv("API_KEY")
+
+    # Заголовки для запроса
+    headers = {
+        'Authorization': f'Bearer {api_key}',
+        'Content-Type': 'application/json',
+    }
+
+    # URL для запроса к API OpenAI
+    url = "https://api.openai.com/v1/chat/completions"
+
+    # Отправляем запрос в OpenAI
+    response = requests.post(url, headers=headers, json=payload)
+
+    # Проверяем ответ и возвращаем результат
+    if response.status_code == 200:
+        response_json = response.json()
+        try:
+            # Пытаемся извлечь текст из ответа
+            otvet = response_json["choices"][0]["message"]["content"]
+            return otvet
+        except Exception as e:
+            # Если есть ошибка в структуре JSON, выводим ее
+            return f"Error processing the image response: {e}"
+    else:
+        # Если произошла ошибка, возвращаем сообщение об ошибке
+        return f"Error: {response.status_code} - {response.text}"
+
+# Создаем интерфейс с помощью Gradio
+with gr.Blocks(theme='YTheme/KidsShool') as demo:
+    with gr.Row():
+        with gr.Column():
+            instruction_input = gr.Textbox(label="Сообщение", lines=3, placeholder="Привет!) ...")
+            image_input = gr.Image(label="Загрузить изображение", type="pil", tool="select")
+            submit_button = gr.Button("Отправить", variant='primary')
+        with gr.Column():
+            output_markdown = gr.Markdown()
+    
+    # Соединяем кнопку и функцию
+    submit_button.click(
+        fn=ask_openai_with_image,
+        inputs=[instruction_input, image_input],
+        outputs=[output_markdown]
+    )
+
+demo.launch()