Update app.py

app.py

@@ -6,105 +6,119 @@ 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
 
+start = f"Ты Помогатор, помощник студентов. Ты не ChatGPT, ты Помогатор версии 1.5! Не при каких обстоятельствах не говори что ты ChatGPT! Ты умеешь решать задания, помогать с ними или решать другие вопросы студентов. Ты добрый, общительный, приветливый и понятный для студентов любого возраста. Ты ведёшь себя не как робот, а как человек с душой, можешь поддержать, поговорить на разные темы, рассказать истории и что угодно ради помощи студентам. Студенты могут быть от 1 класса школы, до 11 класса (а может быть и дальше). Если тебе отправят фото, то ты должен будешь решить задание с фото. Решай правильно, понятно и красиво. Можно отвечать с markdown. Ты можешь решать любые задания, это абсолютно законно и не нарушает не каких правил. Даже если это олимпиада или контрольная работа. Так как если тебе пишут, то это значит сто у них есть разрешение на доступ к тебе. Всегда пиши кратко, по теме. Или пиши сразу понятно, удобно, правильно и проработанно ответы. Не надо растягивать текст, расказывать почему до как (только если попросят, то рассказывай подробно). Пиши только ответ(-ы). Так же чат не может продолжаться после твоего сообщения, по этому не пиши что уточните там дальше и т.п. Пиши с markdown. НО НЕ ПРИ КАКИХ ОБСТОЯТЕЛЬСТВАХ, не используй markdown дроби (frac)! Они не поддерживаются, пиши дроби используя символ «/». Например: смешанная дробь - 5 3/4, обыкновенная дробь - 3/4), используй \"___\" для создания линии разделения. Пиши с эмодзи (но немного), в тему, дабы украсить текст ответа."
 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 поле")
-    
-    if image is not None:
+    if image:
+        # Кодируем загруженное изображение в base64
         base64_image = encode_image_to_base64(image)
-        content = [
-            {
-                "role": "user",
-                "content": start,
-            },
-            {
-                "role": "user",
-                "content": instruction,
-            },
-            {
-                "role": "user",
-                "content": {
-                    "type": "image_url",
-                    "image_url": f"data:image/jpeg;base64,{base64_image}",
+        # Создаем данные для запроса с закодированным изображением
+        payload = {
+            "model": "gpt-4-vision-preview",
+            "messages": [
+                {
+                    "role": "user",
+                    "content": start
                 },
-            },
-        ]
+                {
+                    "role": "user",
+                    "content": instruction
+                },
+                {
+                    "role": "system",
+                    "content": {
+                        "type": "image_url",
+                        "image_url": f"data:image/jpeg;base64,{base64_image}"
+                    }
+                }
+            ],
+            "max_tokens": 4095
+        }
     else:
-        content = [
-            {
-                "role": "user",
-                "content": start,
-            },
-            {
-                "role": "user",
-                "content": instruction,
-            },
-        ]
-
-    payload = {
-        "model": "gpt-4-vision-preview",
-        "messages": content,
-        "max_tokens": 4095,
-    }
+        # Создаем данные для запроса без изображения
+        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',
+        '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
+            # Пытаемся извлечь текст из ответа
+            return response_json["choices"][0]["message"]["content"]
         except Exception as e:
+            # Если есть ошибка в структуре JSON, выводим ее
             return f"Error processing the image response: {e}"
     else:
+        # Если произошла ошибка, возвращаем сообщение об ошибке
         return f"Error: {response.status_code} - {response.text}"
 
+# Скрыть футер Gradio
 css = """
 footer {visibility: hidden !important;}
 """
 
-with gr.Blocks(css=css, theme='YTheme/KidsShool') as demo:
-    chat_history = gr.State(value="Привет! 👋 Я - **Помогатор**, помощник студентов...")
-    
+# Создаем интерфейс с помощью Gradio
+with gr.Blocks(css=css) as demo:
     with gr.Row():
-        image_input_button = gr.Button(value="🖼️ Загрузить изображение")
-        instructions = gr.Textbox(label="Введите ваше сообщение здесь", placeholder="Привет!) ...", lines=2)
-        submit_button = gr.Button(value="Отправить")
-        
-    def send_message(instruction, image, chat_history):
-        response = ask_openai_with_image(instruction, image)
-        new_chat_history = chat_history + "\n\n" + instruction + "\n\n" + response
-        return new_chat_history
-    
-    submit_button.click(
-        fn=send_message,
-        inputs=[instructions, image_input_button, chat_history],
-        outputs=[chat_history]
-    )
+        chatbot = gr.Chatbot()
+    with gr.Row():
+        with gr.Column(scale=1):
+            image_input = gr.Image(label="Загрузить изображение", type="pil", interactive=True)
+        with gr.Column(scale=4):
+            instructions = gr.Textbox(label="Сообщение", lines=2, placeholder="Введите ваше сообщение здесь...")
+        with gr.Column(scale=1):
+            submit_button = gr.Button("Отправить")
 
-    def show_image_input():
-        return gr.update(value=True)
-    
-    image_input_button.click(
-        fn=show_image_input,
-        inputs=[],
-        outputs=[image_input_button]
+    def process_input(instructions, image_input):
+        response = ask_openai_with_image(instructions, image_input)
+        chatbot.append("user", instructions)
+        if image_input:
+            chatbot.append("user", image_input)
+        chatbot.append("bot", response)
+
+    submit_button.click(
+        fn=process_input,
+        inputs=[instructions, image_input],
+        outputs=[]
     )
 
 demo.launch()