Serg4451D commited on
Commit
ad9055e
·
1 Parent(s): 5459fab

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +46 -0
app.py ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import openai
2
+ import streamlit as st
3
+ import os
4
+
5
+ # Установите ваш ключ API OpenAI
6
+ openai.api_key = os.getenv("OPENAI_API_KEY")
7
+
8
+ # Инициализация контекста
9
+ conversation_history = []
10
+
11
+ # Функция для запроса ответа от модели
12
+ def get_model_response(prompt, model, temperature=0.5):
13
+ response = openai.Completion.create(
14
+ engine=model,
15
+ prompt=prompt,
16
+ temperature=temperature,
17
+ max_tokens=1024,
18
+ n=1,
19
+ stop=None,
20
+ frequency_penalty=0,
21
+ presence_penalty=0
22
+ )
23
+
24
+ message = response.choices[0].text.strip()
25
+ conversation_history.append((prompt, message))
26
+ return message
27
+
28
+ # Заголовок приложения
29
+ st.title("Mini-Assistant")
30
+
31
+ # Ввод текста для запроса
32
+ user_input = st.text_input("Ваш вопрос: ", "")
33
+
34
+ # Отправка запроса на сервер OpenAI и получение ответа
35
+ if st.button("Ответить"):
36
+ response = get_model_response(user_input, "text-davinci-002")
37
+
38
+ # Вывод ответа на экран
39
+ st.write(response)
40
+
41
+ # Отображение истории разговора
42
+ if st.checkbox("История разговора"):
43
+ for index, conversation in enumerate(conversation_history):
44
+ st.write(f"Запрос {index + 1}: {conversation[0]}")
45
+ st.write(f"Ответ {index + 1}: {conversation[1]}")
46
+ st.write("-----------")