Salmetov commited on
Commit
7b886ea
·
verified ·
1 Parent(s): 86fa063

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -4
app.py CHANGED
@@ -37,6 +37,9 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
37
  print("Получен запрос:", message)
38
  print("История общения:", history)
39
 
 
 
 
40
  try:
41
  if "район" in message.lower() or "застройщик" in message.lower():
42
  response = get_apartment_data(message)
@@ -46,10 +49,12 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
46
  f"Класс: {row['project_class']}, Площадь: {row['apt_area']}"
47
  for row in response
48
  ])
 
49
  history.append({"role": "user", "content": message})
50
  history.append({"role": "assistant", "content": result})
51
  yield result, history
52
  else:
 
53
  history.append({"role": "user", "content": message})
54
  history.append({"role": "assistant", "content": "Извините, подходящих вариантов не найдено."})
55
  yield "Извините, подходящих вариантов не найдено.", history
@@ -61,17 +66,20 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
61
  messages.append({"role": "user", "content": message})
62
 
63
  response = ""
64
- for message in client.chat_completion(
65
  messages,
66
  max_tokens=max_tokens,
67
  stream=True,
68
  temperature=temperature,
69
  top_p=top_p,
70
  ):
71
- token = message.choices[0].delta.content
72
  response += token
73
- history.append({"role": "assistant", "content": token})
74
- yield response, history
 
 
 
75
  except Exception as e:
76
  print(f"Ошибка обработки запроса: {e}")
77
  history.append({"role": "user", "content": message})
 
37
  print("Получен запрос:", message)
38
  print("История общения:", history)
39
 
40
+ if not history:
41
+ history = []
42
+
43
  try:
44
  if "район" in message.lower() or "застройщик" in message.lower():
45
  response = get_apartment_data(message)
 
49
  f"Класс: {row['project_class']}, Площадь: {row['apt_area']}"
50
  for row in response
51
  ])
52
+ # Обновляем историю в формате Gradio
53
  history.append({"role": "user", "content": message})
54
  history.append({"role": "assistant", "content": result})
55
  yield result, history
56
  else:
57
+ # Если ничего не найдено
58
  history.append({"role": "user", "content": message})
59
  history.append({"role": "assistant", "content": "Извините, подходящих вариантов не найдено."})
60
  yield "Извините, подходящих вариантов не найдено.", history
 
66
  messages.append({"role": "user", "content": message})
67
 
68
  response = ""
69
+ for message_chunk in client.chat_completion(
70
  messages,
71
  max_tokens=max_tokens,
72
  stream=True,
73
  temperature=temperature,
74
  top_p=top_p,
75
  ):
76
+ token = message_chunk.choices[0].delta.content
77
  response += token
78
+
79
+ # Обновляем историю с ответом
80
+ history.append({"role": "user", "content": message})
81
+ history.append({"role": "assistant", "content": response})
82
+ yield response, history
83
  except Exception as e:
84
  print(f"Ошибка обработки запроса: {e}")
85
  history.append({"role": "user", "content": message})