Update app.py

app.py

@@ -15,53 +15,62 @@ def encode_image_to_base64(image):
 
 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:
+        return None
+    if image != None:
         # Кодируем загруженное изображение в base64
         base64_image = encode_image_to_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}"
-                    }
+                    "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
+            "max_tokens": 4095,
         }
-    else:
-        # Создаем данные для запроса без изображения
+
+    if image == None:
+        # Создаем данные для запроса с закодированным изображением
         payload = {
             "model": "gpt-4-vision-preview",
             "messages": [
                 {
                     "role": "user",
-                    "content": start
+                    "content": start,
                 },
                 {
                     "role": "user",
-                    "content": instruction
+                    "content": instruction,
                 }
             ],
-            "max_tokens": 4095
-        }
+            "max_tokens": 4095,
+}    
+
 
     # API ключ для OpenAI
     api_key = os.getenv("API_KEY")
@@ -69,7 +78,7 @@ def ask_openai_with_image(instruction, image):
     # Заголовки для запроса
     headers = {
         'Authorization': f'Bearer {api_key}',
-        'Content-Type': 'application/json'
+        'Content-Type': 'application/json',
     }
 
     # URL для запроса к API OpenAI
@@ -83,40 +92,47 @@ def ask_openai_with_image(instruction, image):
         response_json = response.json()
         try:
             # Пытаемся извлечь текст из ответа
-            return response_json["choices"][0]["message"]["content"]
+            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}"
-
-# Скрыть футер Gradio
 css = """
 footer {visibility: hidden !important;}
 """
-
 # Создаем интерфейс с помощью Gradio
-with gr.Blocks(css=css) as demo:
-    chatbot = gr.Chatbot()
+with gr.Blocks(css=css, theme='YTheme/KidsShool') as demo:
     with gr.Row():
-        image_input = gr.Image(label="Загрузить изображение", type="pil", interactive=True)
-        instructions = gr.Textbox(label="Сообщение", lines=2, placeholder="Введите ваше сообщение здесь...")
-        submit_button = gr.Button("Отправить")
+        with gr.Column():
+            image_input = gr.Image(label="Фото", type="pil")
+            instructions = gr.Textbox(label="Сообщение", lines=3, placeholder="Привет!) ...")
+            submit_button = gr.Button("Отправить", variant='primary')
+            gr.Markdown("🔗 Доступ предоставлен [ЭТИМ](https://ai-hub.rf.gd) сайтом!")
+        with gr.Column():
+            output_markdown = gr.Markdown("""# Привет! 👋
 
-    def process_input(instructions, image_input):
-        response = ask_openai_with_image(instructions, image_input)
-        if instructions:
-            chatbot.add_message(instructions, is_user=True)
-        if image_input:
-            chat_image = Image.open(image_input)
-            chatbot.add_message(chat_image, is_user=True)
-        chatbot.add_message(response, is_user=False)
+Я - **Помогатор**, помощник студентов, специализирующийся на решении разнообразных заданий. Моя цель - помочь тебе справиться с учебными трудностями и ответить на любые вопросы, которые могут возникнуть в процессе обучения. 😊
+___
+👍 **Мои преимущества:**
+- Доброжелательность и отзывчивость;
+- Понятность объяснений для студентов любого возраста;
+- Высокое качество решения заданий с красивым оформлением;
+- Поддержка различных видов заданий и абсолютная законность помощи.
+___
+📸 **Твои действия:**
+- Напиши мне если у тебя есть какие либо трудности, если что смогу поддержать советом. 😊
+- Если ты затрудняешься с каким либо заданием, то просто пришли мне фотографию задания, и я постараюсь максимально помочь, понятно и красиво. ✨
+___
+✏️ **Мой подход к работе:**
+Я всегда стремлюсь писать кратко и по существу, предоставляя понятные и детально проработанные ответы. 🤗""", label="AI Response")
 
     submit_button.click(
-        fn=process_input,
+        fn=ask_openai_with_image,
         inputs=[instructions, image_input],
-        outputs=[]
+        outputs=[output_markdown]
     )
-
 demo.launch()