Update app.py

app.py

@@ -1,25 +1,27 @@
+import logging
+logging.basicConfig(filename="debug.log", level=logging.DEBUG, format='%(asctime)s %(message)s')
 import gradio as gr
 from transformers import pipeline
 
 generator = pipeline("text-generation", model="microsoft/DialoGPT-small")
 
 def chat_fn(message, history=None):
-    print(f"User message: {message}")
+    logging.debug(f"User message: {message}")
     if history is None:
         history = []
     conversation = ""
     for user_msg, bot_msg in history:
         conversation += f"User: {user_msg}\nBot: {bot_msg}\n"
     conversation += f"User: {message}\nBot:"
-    print(f"Built conversation prompt:\n{conversation}")
+    logging.debug(f"Built conversation prompt:\n{conversation}")
 
     try:
         response = generator(conversation, max_length=200, pad_token_id=50256)
         bot_reply = response[0]['generated_text'].split("Bot:")[-1].strip()
-        print(f"Bot reply: {bot_reply}")
+        logging.debug(f"Bot reply: {bot_reply}")
     except Exception as e:
         bot_reply = f"Error generating response: {e}"
-        print(f"Error: {e}")
+        logging.debug(f"Error: {e}")
 
     history.append((message, bot_reply))
     return bot_reply, history