Update app.py

app.py

@@ -1,56 +1,53 @@
+import os
 import gradio as gr
 import requests
-import os
 
-OPENAI_API_KEY = os.getenv("API_KEY")
-API_URL = os.getenv("BASE_URL")
+# Замените на свои значения
+base_url = os.getenv("BASE_URL")
+api_key = os.getenv("API_KEY")
 
-# Список для хранения сообщений в чате
-chat_history = []
+# URL для запроса к GPT-3.5-turbo
+gpt_url = {base_url}
 
-# Функция для генерации ответов на основе диалога
-def generate_response(dialog):
+# Функция для генерации ответа от GPT-3.5-turbo
+def generate_response(prompt):
     headers = {
-        'Content-Type': 'application/json',
-        'Authorization': f'Bearer {OPENAI_API_KEY}',
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}",
     }
 
     data = {
-        'model': 'gpt-3.5-turbo',
-        'messages': [{'role': 'system', 'content': 'You are a helpful assistant.'}] + dialog,
-        'max_tokens': 150,
+        "prompt": prompt,
+        "max_tokens": 150,
     }
 
-    response = requests.post(API_URL, json=data, headers=headers)
-    response_data = response.json()
-    generated_text = response_data['choices'][0]['message']['content'].strip()
-
-    return generated_text
-
-# Функция для обновления чата
-def update_chat(input_text):
-    user_message = {'role': 'user', 'content': input_text}
-    chat_history.append(user_message)
-    
-    # Генерация ответа на основе текущего диалога
-    assistant_reply = generate_response(chat_history)
-    assistant_message = {'role': 'assistant', 'content': assistant_reply}
-    chat_history.append(assistant_message)
-    
-    # Обновление текста в поле вывода
-    output_text = "Chat History:\n"
-    for message in chat_history:
-        output_text += f"{message['role']}: {message['content']}\n"
-
-    return output_text
-
-# Интерфейс Gradio для чата
-iface = gr.Interface(
-    fn=update_chat,
-    inputs=gr.Textbox(),
-    outputs=gr.Textbox(),
-    live=True,
-    title="Chat with GPT-3.5-turbo"
-)
-
-iface.launch()
+    response = requests.post(gpt_url, json=data, headers=headers)
+    return response.json()["choices"][0]["text"].strip()
+
+# Создание интерфейса чата с использованием Gradio
+def chat_interface(messages):
+    # Собираем все сообщения в одну строку, чтобы использовать как контекст для GPT-3.5-turbo
+    conversation_history = "\n".join(messages)
+
+    # Генерируем ответ от GPT-3.5-turbo
+    response = generate_response(conversation_history)
+
+    # Добавляем сгенерированный ответ к списку сообщений
+    messages.append(response)
+
+    # Создаем интерфейс Gradio с возможностью удаления или изменения каждого сообщения
+    iface = gr.Interface(
+        fn=chat_interface,
+        inputs=["text"],
+        outputs="text",
+        live=True,
+        examples=[[conversation_history]],
+        title="Chat with GPT-3.5-turbo",
+        description="Type a message and see GPT-3.5-turbo's response.",
+    )
+
+    return iface
+
+# Запуск интерфейса Gradio
+if __name__ == "__main__":
+    chat_interface([]).launch()