Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import json
 import os
+import time
 
 def generate(description, model, max_tokens):
     headers = {
@@ -15,21 +16,17 @@ def generate(description, model, max_tokens):
         'model': model
     }
 
-    response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
-    data = json.loads(response.text)
+    while True:
+        response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
+        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}'
-    else:
-        return f'Не удалось сгенерировать текст. {data}'
-
-iface = gr.Interface(fn=generate, inputs=[
-    gr.Textbox(label="Запрос"),
-    gr.Radio(show_label=True, label="Модель", interactive=True, choices=["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4"], value="gpt-3.5-turbo"),
-    gr.Slider(show_label=True, label="Максимум токенов", minimum=100, maximum=15000, value=10000, step=1)
-], outputs=gr.Textbox(label="Ответ"), title="EasyGPT")
-iface.launch()
+        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']
+            print(f'Ошибка: {error_message}')
+            time.sleep(1)
+        else:
+            print(f'Не удалось сгенерировать текст. Повторная попытка...')
+            time.sleep(1)