Update app.py

app.py

@@ -1,14 +1,18 @@
 import gradio as gr
 import requests
 import os
+import base64
 from PIL import Image
-import io
+import numpy as np
 
 # Функция для обработки изображения и текста и отправки запроса к OpenAI
 def generate_text(image, prompt):
+    # Конвертируем NumPy массив в объект Image
+    image_pil = Image.fromarray(np.uint8(image)).convert('RGB')
+    
     # Конвертируем изображение для отправки через API
     image_bytes = io.BytesIO()
-    image.save(image_bytes, format='PNG')
+    image_pil.save(image_bytes, format='PNG')
     image_base64 = base64.b64encode(image_bytes.getvalue()).decode('utf-8')
 
     # API ключ для OpenAI
@@ -34,7 +38,7 @@ def generate_text(image, prompt):
     }
 
     # Отправляем запрос к OpenAI
-    response = requests.post('https://api.openai.com/v1/completions', headers=headers, json=data)
+    response = requests.post('https://api.openai.com/v1/engines/davinci-codex/completions', headers=headers, json=data)
     
     # Проверяем ответ и возвращаем результат
     if response.status_code == 200:
@@ -47,7 +51,7 @@ def generate_text(image, prompt):
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            image_input = gr.Image(label="Загрузите изображение")
+            image_input = gr.Image(label="Загрузите изображение", type="numpy")
             text_input = gr.Textbox(label="Введите текст")
             submit_button = gr.Button("Решить")
         with gr.Column():