Rooni commited on
Commit
fe8d2aa
·
verified ·
1 Parent(s): f4c1707

Update app.py

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