Update app.py

app.py

@@ -5,7 +5,8 @@ import os
 OPENAI_API_KEY = os.getenv("API_KEY")
 API_URL = os.getenv("BASE_URL")
 
-def generate_response(prompt):
+# Функция для генерации ответов на основе диалога
+def generate_response(dialog):
     headers = {
         'Content-Type': 'application/json',
         'Authorization': f'Bearer {OPENAI_API_KEY}',
@@ -13,21 +14,23 @@ def generate_response(prompt):
 
     data = {
         'model': 'gpt-3.5-turbo',
-        'prompt': prompt,
+        'messages': [{'role': 'system', 'content': 'You are a helpful assistant.'}] + dialog,
         'max_tokens': 150,
     }
 
     response = requests.post(API_URL, json=data, headers=headers)
     response_data = response.json()
-    generated_text = response_data['choices'][0]['text'].strip()
+    generated_text = response_data['choices'][0]['message']['content'].strip()
 
     return generated_text
 
 iface = gr.Interface(
     fn=generate_response,
-    inputs=gr.Textbox(),
+    inputs=gr.Textbox(prompt="You:"),
     outputs=gr.Textbox(),
     live=True,
+    title="Chat with GPT-3.5-turbo",
+    layout="vertical",
 )
 
 iface.launch()