Update app.py

app.py

@@ -16,13 +16,14 @@ params = {
 
 def save_log(task, to_save):
   with open("logs.txt", "a") as log_file:
-    current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    log_file.write(f"[{current_time}] - {task}: {to_save}\n")
+      current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+      log_file.write(f"[{current_time}] - {task}: {to_save}\n")
+      print(to_save)
         
 
 llm = AutoModelForCausalLM.from_pretrained("Aspik101/Llama-2-7b-chat-hf-pl-lora_GGML", model_type="llama")
 
-with gr.Blocks(allow_flagging="auto") as demo:
+with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
     clear = gr.Button("Clear")
@@ -41,6 +42,7 @@ with gr.Blocks(allow_flagging="auto") as demo:
             time.sleep(0.005)
             yield history
 
+        save_log("answer", answer_save)
     msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
         bot, chatbot, chatbot
     )