Update app.py

app.py

@@ -1,58 +1,55 @@
-import os
 import gradio as gr
 import requests
+import os
 
-# Замените на свои значения
-base_url = os.getenv("BASE_URL")
-api_key = os.getenv("API_KEY")
-
-# URL для запроса к GPT-3.5-turbo
-gpt_url = f'{base_url}'
+BASE_URL = os.getenv("BASE_URL")
+API_KEY = os.getenv("API_KEY")
 
-# Функция для генерации ответа от GPT-3.5-turbo
 def generate_response(prompt):
+    endpoint = f'{BASE_URL}''
     headers = {
         "Content-Type": "application/json",
-        "Authorization": f"Bearer {api_key}",
+        "Authorization": f"Bearer {API_KEY}",
     }
-
     data = {
-        "inputs": prompt,
+        "model": "gpt-3.5-turbo",  # Используйте "text-gpt-3.5-turbo" для GPT-3.5-turbo
+        "prompt": prompt,
+        "temperature": 0.7,
         "max_tokens": 150,
     }
 
-    response = requests.post(gpt_url, json=data, headers=headers)
-    
-    # Проверяем наличие ключа 'choices' в ответе
-    if "choices" in response.json():
-        return response.json()["choices"][0]["text"].strip()
-    else:
-        return "Error: Unable to retrieve response."
-
-# Создание интерфейса чата с использованием Gradio
-def chat_interface(messages):
-    # Собираем все сообщения в одну строку, чтобы использовать как контекст для GPT-3.5-turbo
-    conversation_history = "\n".join(messages)
-
-    # Генерируем ответ от GPT-3.5-turbo
-    response = generate_response(conversation_history)
-
-    # Обновляем список сообщений с новым ответом
-    updated_messages = messages + [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, updated_messages
-
-# Запуск интерфейса Gradio
-if __name__ == "__main__":
-    chat_interface([])[0].launch()
+    response = requests.post(endpoint, headers=headers, json=data)
+    return response.json()["choices"][0]["text"]
+
+# Определение функций для взаимодействия с чатом
+def send_message(messages, new_message):
+    messages.append(new_message)
+    return messages
+
+def delete_message(messages, index):
+    if 0 <= index < len(messages):
+        del messages[index]
+    return messages
+
+def edit_message(messages, index, new_message):
+    if 0 <= index < len(messages):
+        messages[index] = new_message
+    return messages
+
+iface = gr.Interface(
+    fn=generate_response,
+    inputs=["text"],
+    outputs="text",
+    live=True,
+    theme="compact",
+)
+
+# Настройка UI для чата с возможностью удаления и изменения сообщений
+iface.add_textbox("Введите сообщение:", key="input_message")
+iface.add_button("Отправить", lambda input_message: send_message(iface.data.input_message, input_message), key="send_button")
+iface.add_textbox("Удалить сообщение (индекс):", key="delete_index")
+iface.add_button("Удалить", lambda delete_index: delete_message(iface.data.input_message, int(delete_index)), key="delete_button")
+iface.add_textbox("Изменить сообщение (индекс, новое сообщение):", key="edit_params")
+iface.add_button("Изменить", lambda edit_params: edit_message(iface.data.input_message, *map(str.strip, edit_params.split(','))), key="edit_button")
+
+iface.launch()