Update app.py

app.py

@@ -1,29 +1,31 @@
 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):
-    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:"
-    logging.debug(f"Built conversation prompt:\n{conversation}")
-
     try:
+        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:"
+        logging.debug(f"Prompt:\n{conversation}")
+
         response = generator(conversation, max_length=200, pad_token_id=50256)
         bot_reply = response[0]['generated_text'].split("Bot:")[-1].strip()
         logging.debug(f"Bot reply: {bot_reply}")
-    except Exception as e:
-        bot_reply = f"Error generating response: {e}"
-        logging.debug(f"Error: {e}")
 
-    history.append((message, bot_reply))
-    return bot_reply, history
+        history.append((message, bot_reply))
+        return bot_reply, history
+
+    except Exception as e:
+        logging.error(f"Exception in chat_fn: {e}", exc_info=True)
+        return f"Oops, something went wrong: {e}", history or []
 
 gr.ChatInterface(fn=chat_fn).launch()