Update app.py

app.py

@@ -217,7 +217,8 @@ async def chat(query,history,sources,reports,subtype, client_ip=None, session_id
                     answer_yet += token
                     parsed_answer = parse_output_llm_with_sources(answer_yet)
                     history[-1] = (query, parsed_answer)
-                yield [tuple(x) for x in history], docs_html
+                    logs_data["answer"] = parsed_answer
+                    yield [tuple(x) for x in history], docs_html, logs_data, session_id
 
         # Stream the response updates
         async for update in process_stream():
@@ -256,7 +257,7 @@ async def chat(query,history,sources,reports,subtype, client_ip=None, session_id
                 
                 response = chat_model.chat_completion(
                     messages=formatted_messages,
-                    max_tokens=int(model_config.get('reader', 'MAX_TOKENS'))
+                    max_tokens= int(model_config.get('reader', 'MAX_TOKENS'))
                 )
                 
                 response_text = response.choices[0].message.content
@@ -278,9 +279,9 @@ async def chat(query,history,sources,reports,subtype, client_ip=None, session_id
 
     # logging the event
     try:
-        save_logs(scheduler,JSON_DATASET_PATH,logs)
+        save_logs(scheduler,JSON_DATASET_PATH,logs_data)
     except Exception as e:
-        logging.error(e)
+        raise