vortex123 commited on
Commit
67219c7
·
verified ·
1 Parent(s): f65bf9b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +44 -0
app.py ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from openai import OpenAI
3
+
4
+ # Инициализация клиента DeepSeek
5
+ client = OpenAI(
6
+ api_key="sk-a02694cf3c8640c9ae60428ee2c5a62e", # Замените ключ на свой
7
+ base_url="https://api.deepseek.com"
8
+ )
9
+
10
+ st.title("Чат с deepseek-reasoner")
11
+
12
+ # Храним историю переписки в Session State, чтобы она сохранялась между ререндами
13
+ if "messages" not in st.session_state:
14
+ st.session_state["messages"] = []
15
+
16
+ # Поле ввода для пользователя
17
+ user_input = st.text_input("Ваш вопрос:", "")
18
+
19
+ # По клику на кнопку отправляем запрос к модели
20
+ if st.button("Отправить"):
21
+ if user_input.strip():
22
+ # Добавляем сообщение пользователя в историю
23
+ st.session_state["messages"].append({"role": "user", "content": user_input})
24
+
25
+ try:
26
+ # Отправляем весь контекст (историю) модели deepseek-reasoner
27
+ response = client.chat.completions.create(
28
+ model="deepseek-reasoner",
29
+ messages=st.session_state["messages"]
30
+ )
31
+ assistant_message = response.choices[0].message["content"]
32
+
33
+ # Сохраняем ответ ассистента в историю
34
+ st.session_state["messages"].append(
35
+ {"role": "assistant", "content": assistant_message}
36
+ )
37
+ except Exception as e:
38
+ st.error(f"Ошибка при обращении к API: {e}")
39
+
40
+ # Выводим чат в формате "Роль: Текст"
41
+ st.write("---")
42
+ for msg in st.session_state["messages"]:
43
+ role = "Вы" if msg["role"] == "user" else "Модель"
44
+ st.markdown(f"**{role}:** {msg['content']}")