Update app.py

app.py

@@ -18,118 +18,93 @@ 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 != None:
-        # Кодируем загруженное изображение в base64
+    
+    if image is not None:
         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,
-        }
-
-    if image == None:
-        # Создаем данные для запроса с закодированным изображением
-        payload = {
-            "model": "gpt-4-vision-preview",
-            "messages": [
-                {
-                    "role": "user",
-                    "content": start,
+        content = [
+            {
+                "role": "user",
+                "content": start,
+            },
+            {
+                "role": "user",
+                "content": instruction,
+            },
+            {
+                "role": "user",
+                "content": {
+                    "type": "image_url",
+                    "image_url": f"data:image/jpeg;base64,{base64_image}",
                 },
-                {
-                    "role": "user",
-                    "content": instruction,
-                }
-            ],
-            "max_tokens": 4095,
-}    
+            },
+        ]
+    else:
+        content = [
+            {
+                "role": "user",
+                "content": start,
+            },
+            {
+                "role": "user",
+                "content": instruction,
+            },
+        ]
 
+    payload = {
+        "model": "gpt-4-vision-preview",
+        "messages": content,
+        "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"]
-            print(f"Генерация \"{instruction}\": {otvet}")
             return otvet
         except Exception as e:
-            # Если есть ошибка в структуре JSON, выводим ее
             return f"Error processing the image response: {e}"
     else:
-        # Если произошла ошибка, возвращаем сообщение об ошибке
         return f"Error: {response.status_code} - {response.text}"
-        
+
 css = """
 footer {visibility: hidden !important;}
 """
 
 with gr.Blocks(css=css, theme='YTheme/KidsShool') as demo:
-    chat_history = gr.Textbox(label="История чата", lines=20, interactive=False, value="Привет! 👋 Я - **Помогатор**, помощник студентов...")
+    chat_history = gr.State(value="Привет! 👋 Я - **Помогатор**, помощник студентов...")
     
     with gr.Row():
-        image_input = gr.Image(label="Загрузить изображение", type="pil", tool="upload", visible=False)
+        image_input_button = gr.Button(value="🖼️ Загрузить изображение")
         instructions = gr.Textbox(label="Введите ваше сообщение здесь", placeholder="Привет!) ...", lines=2)
         submit_button = gr.Button(value="Отправить")
         
-    def send_message(instruction, image):
+    def send_message(instruction, image, chat_history):
         response = ask_openai_with_image(instruction, image)
-        chat_history.update(value=chat_history.value + "\n\n" + instruction + "\n\n" + response)
+        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],
-        outputs=[]
+        inputs=[instructions, image_input_button, chat_history],
+        outputs=[chat_history]
     )
 
     def show_image_input():
-        image_input.visible = True
+        return gr.update(value=True)
     
-    image_input_button = gr.Button(value="🖼️")
     image_input_button.click(
         fn=show_image_input,
         inputs=[],
-        outputs=[]
+        outputs=[image_input_button]
     )
 
 demo.launch()