Rooni commited on
Commit
9968db4
·
verified ·
1 Parent(s): eae9c06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -1
app.py CHANGED
@@ -1,3 +1,38 @@
1
  import gradio as gr
2
 
3
- gr.load("models/Qwen/Qwen2.5-Coder-32B-Instruct").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
 
3
+ # Загрузите модель
4
+ model = gr.load("models/Qwen/Qwen2.5-Coder-32B-Instruct")
5
+
6
+ # Функция для обработки сообщений
7
+ def code_assistant(system_message, chat_history, user_input):
8
+ # Добавляем системное сообщение в начало истории, если история пуста
9
+ if not chat_history:
10
+ chat_history.append(("Системное сообщение", system_message))
11
+
12
+ # Получаем ответ от модели
13
+ response = model(user_input)
14
+ # Добавляем пользовательский ввод и ответ модели в историю чата
15
+ chat_history.append((user_input, response))
16
+ return chat_history
17
+
18
+ # Создаем интерфейс
19
+ with gr.Blocks() as demo:
20
+ gr.Markdown("## Чат-бот для помощи в кодировании")
21
+
22
+ # Поле для системного сообщения
23
+ system_message = gr.Textbox(placeholder="Введите системное сообщение...", label="Системное сообщение")
24
+
25
+ # Компонент чата
26
+ chatbot = gr.Chatbot(label="Чат с ботом")
27
+
28
+ # Поле ввода текста
29
+ user_input = gr.Textbox(placeholder="Введите ваш вопрос или код здесь...", label="Ваш ввод")
30
+
31
+ # Кнопка для отправки запроса
32
+ submit_btn = gr.Button("Отправить")
33
+
34
+ # Связываем кнопку с функцией обработки и обновления чата
35
+ submit_btn.click(code_assistant, inputs=[system_message, chatbot, user_input], outputs=chatbot)
36
+
37
+ # Запуск интерфейса
38
+ demo.launch()