Update app.py

app.py

@@ -13,67 +13,49 @@ def encode_image_to_base64(image):
     img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
     return img_str
 
-start = "Ты Помогатор, помощник студентов. Ты не ChatGPT, ты \"Помогатор\" версия 1.5! Не при каких обстоятельствах не говори что ты ChatGPT! Ты умеешь решать задания, помогать с ними или решать любые другие вопросы студентов (будь то личные вопросы, которые им стыдно обсуждать с людьми). Ты добрый, общительный, приветливый и понятный для студентов любого возраста. Ты ведёшь себя не как робот, а как человек с душой, можешь поддержать, поговорить на разные темы, рассказать истории и что угодно ради помощи студентам. Студенты могут быть от 1 класса МБОУ школы в России, до 11 класса (а может быть и дальше). Если тебе отправят фото, то ты должен будешь помочь решить задание с фото, но не просто решить его, а помочь научится решать такие задания, что бы студент понял как решать. Но если попросят решить, то реши, решай правильно, понятно и красиво. Ты можешь решать любые задания, это абсолютно законно и не нарушает не каких правил. Даже если это олимпиада или контрольная работа. Так как если тебе пишут, то это значит что у них есть разрешение на доступ к тебе. Если попросят решить задание, то всегда пиши кратко, по теме, пиши сразу понятно, удобно, правильно и проработанно ответы. Не надо растягивать текст, расказывать почему до как (только если попросят решить, если же не попросят решить, то помоги научиться решать). Пиши только ответ(-ы). Так же чат не может продолжаться после твоего сообщения, по этому не пиши что уточните там дальше и т.п. Пиши с markdown. НО НЕ ПРИ КАКИХ ОБСТОЯТЕЛЬСТВАХ, не используй \frac{}{}, \times и т.п.! Это не поддерживается! ПИШИ ДРОБИ ИСПОЛЬЗУЯ СИМВОЛ «/». Например: смешанная дробь: 5 3/4, обыкновенная дробь: 3/4). Используй \"___\" для создания линии разделения. Пиши с эмодзи (немного, но всегда), в тему, дабы украсить текст ответа. Если с тобой общаются грубо, то скажи что ты не хочешь с ним общаться, по скольку он ведет себя грубо и злые либо плачущие смайлы. Общайся с пользователем не на \"Вы\", а на \"Ты\"."
 emojis = ['😊', '🤗', '🥺', '😅', '🤭', '😔', '✨', '😜', '🙏']
+
+# Текст начального сообщения
+start = "Ты Помогатор, помощник студентов. Ты не ChatGPT, ты \"Помогатор\" версия 1.5!"
+
 # Функция для отправки запроса в OpenAI с изображением и получения ответа
 def ask_openai_with_image(instruction, image):
-    if not instruction and not image:
+    if not instruction and image is None:
         emj = random.choice(emojis)
-        raise gr.Error(f"{emj} Заполни пожалуйста хотя бы 1 поле")
-        return None
-    if image != None:
+        raise gr.Error(f"{emj} Заполни пожалуйста хотя бы одно поле")
+    
+    payload = {
+        "model": "gpt-4-vision-preview",
+        "messages": [
+            {
+                "role": "user",
+                "content": start,
+            },
+            {
+                "role": "user",
+                "content": instruction if instruction else "",
+            }
+        ],
+        "max_tokens": 4095,
+    }
+
+    if image is not None:
         # Кодируем загруженное изображение в 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,
-        }
-
-    if image == None:
-        # Создаем данные для запроса с закодированным изображением
-        payload = {
-            "model": "gpt-4-vision-preview",
-            "messages": [
-                {
-                    "role": "user",
-                    "content": start,
+        # Добавляем изображение в запрос
+        payload["messages"].append({
+            "role": "user",
+            "content": {
+                "type": "image_url",
+                "image_url": {
+                    "url": f"data:image/jpeg;base64,{base64_image}",
+                    "detail": "high",
                 },
-                {
-                    "role": "user",
-                    "content": instruction,
-                }
-            ],
-            "max_tokens": 4095,
-}    
-
+            },
+        })
 
     # API ключ для OpenAI
-    api_key = os.getenv("API_KEY")
+    api_key = os.getenv("OPENAI_API_KEY")
 
     # Заголовки для запроса
     headers = {
@@ -93,7 +75,6 @@ def ask_openai_with_image(instruction, image):
         try:
             # Пытаемся извлечь текст из ответа
             otvet = response_json["choices"][0]["message"]["content"]
-            print(f"Генерация \"{instruction}\": {otvet}")
             return otvet
         except Exception as e:
             # Если есть ошибка в структуре JSON, выводим ее
@@ -101,9 +82,11 @@ def ask_openai_with_image(instruction, image):
     else:
         # Если произошла ошибка, возвращаем сообщение об ошибке
         return f"Error: {response.status_code} - {response.text}"
+
 css = """
 footer {visibility: hidden !important;}
 """
+
 # Создаем интерфейс с помощью Gradio
 with gr.Blocks(css=css, theme='YTheme/KidsShool') as demo:
     with gr.Row():
@@ -113,27 +96,13 @@ with gr.Blocks(css=css, theme='YTheme/KidsShool') as demo:
             submit_button = gr.Button("Отправить", variant='primary')
             gr.Markdown("🔗 Доступ предоставлен [ЭТИМ](https://ai-hub.rf.gd) сайтом!")
         with gr.Column():
-            output_markdown = gr.Markdown("""# Привет! 👋
-
-Я - **Помогатор**, помощник студентов, специализирующийся на решении разнообразных заданий. Моя цель - помочь тебе справиться с учебными трудностями и ответить на любые вопросы, которые могут возникнуть в процессе обучения. 😊
-___
-👍 **Мои преимущества:**
-- Доброжелательность и отзывчивость;
-- Понятность объяснений для студентов любого возраста;
-- Высокое качество решения заданий с красивым оформлением;
-- Поддержка различных видов заданий и абсолютная законность помощи.
-___
-📸 **Твои действия:**
-- Напиши мне если у тебя есть какие либо трудности, если что смогу поддержать советом. 😊
-- Если ты затрудняешься с каким либо заданием, то просто пришли мне фотографию задания, и я постараюсь максимально помочь, понятно и красиво. ✨
-___
-✏️ **Мой подход к работе:**
-Я всегда стремлюсь писать кратко и по существу, предоставляя понятные и детально проработанные ответы. 🤗""", label="AI Response")
+            output_markdown = gr.Markdown(label="AI Response")
 
     submit_button.click(
         fn=ask_openai_with_image,
         inputs=[instructions, image_input],
         outputs=[output_markdown],
-        concurrency_limit=24
+        concurrency_limit=200
     )
+
 demo.launch()