lazarus commited on
Commit
3ca52be
·
verified ·
1 Parent(s): 6d0d079

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -2
app.py CHANGED
@@ -12,7 +12,30 @@ def load_model():
12
 
13
  # Ініціалізація моделі
14
  generator = load_model()
 
15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  # Функція для генерації відповіді
17
  def chat_with_npc(user_input):
18
  if isinstance(generator, str): # Якщо модель не завантажилась
@@ -36,7 +59,7 @@ interface = gr.Interface(
36
  fn=chat_with_npc,
37
  inputs="text",
38
  outputs="text",
39
- title="NPC Помічник",
40
  description="Оптимізований віртуальний помічник для запитів з великою кількістю слів.",
41
  examples=[
42
  ["Як можна розвивати духовний потенціал у сучасному світі?"],
@@ -46,4 +69,5 @@ interface = gr.Interface(
46
 
47
  # Запуск програми
48
  if __name__ == "__main__":
49
- interface.launch()
 
 
12
 
13
  # Ініціалізація моделі
14
  generator = load_model()
15
+ conversation_history = []
16
 
17
+ def chat_with_npc(user_input):
18
+ global conversation_history
19
+ if isinstance(generator, str):
20
+ return generator
21
+ try:
22
+ # Додаємо новий запит до історії
23
+ conversation_history.append({"role": "user", "content": user_input})
24
+
25
+ # Формуємо контекст
26
+ context = "\n".join([f"{msg['role']}: {msg['content']}" for msg in conversation_history])
27
+
28
+ # Генеруємо відповідь
29
+ response = generator(context, max_length=300, num_return_sequences=1, do_sample=True, temperature=0.7, top_p=0.9)
30
+ bot_reply = response[0]["generated_text"]
31
+
32
+ # Додаємо відповідь до історії
33
+ conversation_history.append({"role": "assistant", "content": bot_reply})
34
+ return bot_reply
35
+ except Exception as e:
36
+ return f"Помилка генерації відповіді: {e}"
37
+
38
+
39
  # Функція для генерації відповіді
40
  def chat_with_npc(user_input):
41
  if isinstance(generator, str): # Якщо модель не завантажилась
 
59
  fn=chat_with_npc,
60
  inputs="text",
61
  outputs="text",
62
+ title="Alt Помічник",
63
  description="Оптимізований віртуальний помічник для запитів з великою кількістю слів.",
64
  examples=[
65
  ["Як можна розвивати духовний потенціал у сучасному світі?"],
 
69
 
70
  # Запуск програми
71
  if __name__ == "__main__":
72
+ interface.launch()
73
+