Update app.py

app.py

@@ -1,55 +1,33 @@
 import gradio as gr
-import requests
-import json
+import openai
 import os
-import numpy as np
-from PIL import Image
-from io import BytesIO
 
-def generate_minecraft_command(description, image):
-    try:
-        headers = {
-            'Authorization': f'Bearer {os.getenv("API_KEY")}',
-            'Content-Type': 'application/json'
-        }
+# Установите свои учетные данные API от OpenAI
+openai.api_key = os.getenv("API_KEY")
 
-        # Преобразуем изображение в формат bytes
-        image_bytes = BytesIO()
-        image.save(image_bytes, format='PNG')
-        image_data = image_bytes.getvalue()
+# Установите базовый URL
+base_url = os.getenv("BASE_URL")
 
-        files = {'image': ('image.png', image_data, 'image/png')}
-
-        data = {
-            'messages': [{'role': 'system', 'content': description}],
-            'max_tokens': 10000,
-            'model': os.getenv("MODEL")
-        }
-
-        response = requests.post(os.getenv("BASE_URL"), headers=headers, json=data, files=files)
-        response.raise_for_status()
-
-        data = json.loads(response.text)
-
-        if 'choices' in data and len(data['choices']) > 0:
-            command = data['choices'][0]['message']['content'].strip()
-            return command
-        elif 'error' in data:
-            error_message = data['error']['message']
-            return f'Ошибка: {error_message}\nДополнительная информация: {data}'
-        else:
-            return f'Не удалось сгенерировать команду. {data}'
-
-    except Exception as e:
-        return f'Произошла ошибка: {str(e)}'
+def generate_solution(prompt):
+    # Используйте GPT-3.5-turbo для генерации решения
+    response = openai.Completion.create(
+        engine="text-davinci-003",  # Выберите движок, поддерживающий GPT-3.5-turbo
+        prompt=prompt,
+        max_tokens=150  # Установите максимальное количество токенов в ответе
+    )
+    return response.choices[0].text.strip()
 
+# Определите интерфейс Gradio
 iface = gr.Interface(
-    fn=generate_minecraft_command,
-    inputs=[
-        gr.Textbox(label="Запрос"),
-        gr.Image(label="Изображение", type="pil")  # Указываем Gradio использовать PIL для обработки изображения
-    ],
-    outputs=gr.Textbox(label="Ответ"),
-    title="GPT"
+    fn=generate_solution,
+    inputs="text",  # Ввод текста
+    outputs="text",  # Вывод текста
+    live=True,
+    theme="huggingface",  # Используйте тему Hugging Face
+    layout="vertical",
+    title="GPT-3.5-turbo Решение Генератор",
+    description="Загрузите фото, введите промпт и нажмите на кнопку 'Решить' для генерации решения.",
 )
-iface.launch()
+
+# Запустите приложение Gradio
+iface.launch(share=True, base_url=base_url)