Update app.py

app.py

@@ -1,46 +1,36 @@
 import openai
 import streamlit as st
-import os
 
-# Установите ваш ключ API OpenAI
-openai.api_key ="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"
+# Установка ключа API OpenAI
+openai.api_key = "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"
 
-# Инициализация контекста
-conversation_history = []
-
-# Функция для запроса ответа от модели
-def get_model_response(prompt, model, temperature=0.5):
+# Функция для генерации ответа
+def generate_answer(prompt, model, max_tokens):
     response = openai.Completion.create(
-      engine=model,
-      prompt=prompt,
-      temperature=temperature,
-      max_tokens=1024,
-      n=1,
-      stop=None,
-      frequency_penalty=0,
-      presence_penalty=0
+        engine=model,
+        prompt=prompt,
+        max_tokens=max_tokens,
+        n=1,
+        stop=None,
+        temperature=0.7,
     )
 
-    message = response.choices[0].text.strip()
-    conversation_history.append((prompt, message))
-    return message
-
-# Заголовок приложения
-st.title("Mini-Assistant")
+    message = response.choices[0].text
+    return message.strip()
 
-# Ввод текста для запроса
-user_input = st.text_input("Ваш вопрос: ", "")
+# Заголовок страницы
+st.title("Привет! Я - ChatGPT")
 
-# Отправка запроса на сервер OpenAI и получение ответа
-if st.button("Ответить"):
-    response = get_model_response(user_input, "text-davinci-002")
+# Поле для ввода сообщения
+message = st.text_input("Введите ваше сообщение:")
 
-    # Вывод ответа на экран
-    st.write(response)
+if message:
+    # Генерация ответа
+    response = generate_answer(
+        prompt=message,
+        model="gpt-3.5-turbo",
+        max_tokens=150,
+    )
 
-# Отображение истории разговора
-if st.checkbox("История разговора"):
-    for index, conversation in enumerate(conversation_history):
-        st.write(f"Запрос {index + 1}: {conversation[0]}")
-        st.write(f"Ответ {index + 1}: {conversation[1]}")
-        st.write("-----------")
+    # Вывод ответа
+    st.write("Ответ:", response)