Prajith04 commited on
Commit
d8d90e3
·
verified ·
1 Parent(s): 4d41769

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -5
main.py CHANGED
@@ -21,6 +21,7 @@ cache_dir = os.environ.get("MODEL_CACHE_DIR", "/app/cache") # Fallback to /app/
21
  os.makedirs(cache_dir, exist_ok=True)
22
  gliner_model = GLiNER.from_pretrained("urchade/gliner_medium-v2.1",cache_dir=cache_dir)
23
  groq_client = Groq(api_key=GROQ_API_KEY)
 
24
 
25
  def extract_entities(text):
26
  # Tokenize the input text first
@@ -94,13 +95,11 @@ async def chat(request: Request, message: str = Form(...)):
94
  bot_reply = generate_response(message, answer)
95
  else:
96
  bot_reply = "Sorry, I couldn't find a suitable answer. Please contact support."
 
 
97
 
98
- chat_history = [
99
- {"sender": "User", "message": message},
100
- {"sender": "Bot", "message": bot_reply}
101
- ]
102
  return templates.TemplateResponse("chat.html", {
103
  "request": request,
104
- "chat_history": chat_history,
105
  "entities": entity_info
106
  })
 
21
  os.makedirs(cache_dir, exist_ok=True)
22
  gliner_model = GLiNER.from_pretrained("urchade/gliner_medium-v2.1",cache_dir=cache_dir)
23
  groq_client = Groq(api_key=GROQ_API_KEY)
24
+ chat_memory=[]
25
 
26
  def extract_entities(text):
27
  # Tokenize the input text first
 
95
  bot_reply = generate_response(message, answer)
96
  else:
97
  bot_reply = "Sorry, I couldn't find a suitable answer. Please contact support."
98
+ chat_memory.append({"sender": "User", "message": message})
99
+ chat_memory.append({"sender": "Bot", "message": bot_reply})
100
 
 
 
 
 
101
  return templates.TemplateResponse("chat.html", {
102
  "request": request,
103
+ "chat_history": chat_memory,
104
  "entities": entity_info
105
  })