Update app.py

app.py

@@ -6,14 +6,15 @@ Automatically generated by Colab.
 Original file is located at
     https://colab.research.google.com/drive/1UgXple_p_R-0mq9p5vhOmFPo9cgdayJy
 """
-
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
+# Load the pretrained DialoGPT model
 model_name = "microsoft/DialoGPT-small"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
+# Hardcoded rule-based responses
 rules = {
     "hi": "Hello! How can I help you today?",
     "hello": "Hi there! How can I assist you?",
@@ -34,19 +35,15 @@ rules = {
     "what is ai": "AI stands for Artificial Intelligence, which is intelligence demonstrated by machines.",
     "who created you": "I was created by a talented developer using Python and machine learning!",
     "how can i learn programming": "Start with basics like Python. There are many free tutorials online to get you started!",
-    'ok': 'ok',
-    'who are you?': 'I am nino',
-    'hi nino': 'hi there',
+    "ok": "ok",
+    "who are you?": "I am nino",
+    "hi nino": "hi there",
 }
 
 def respond(user_input, history):
-    # Defensive type checks
     if not isinstance(history, list):
-        print(f"Warning: Expected list for history but got {type(history)}. Resetting history.")
         history = []
-
     if not isinstance(user_input, str):
-        print(f"Warning: Expected str for user_input but got {type(user_input)}. Coercing to str.")
         user_input = str(user_input)
 
     user_input_clean = user_input.lower().strip()
@@ -79,33 +76,37 @@ def respond(user_input, history):
             bot_reply = "I'm not sure how to respond to that. Can you rephrase it?"
 
     history.append((user_input, bot_reply))
-    return history, history
+    return history, "", history
 
 def save_chat(history):
-    if history is not None:
-        try:
-            with open("/tmp/chat_history.txt", "w", encoding="utf-8") as f:
-                for user_msg, bot_msg in history:
-                    if bot_msg is not None:
-                        f.write(f"You: {user_msg}\nBot: {bot_msg}\n\n")
-                    else:
-                        f.write(f"You: {user_msg}\nBot: (No response)\n\n")
-        except Exception as e:
-            print(f"Error saving chat: {e}")
+    try:
+        with open("/tmp/chat_history.txt", "w", encoding="utf-8") as f:
+            for user_msg, bot_msg in history:
+                if bot_msg is not None:
+                    f.write(f"You: {user_msg}\nBot: {bot_msg}\n\n")
+                else:
+                    f.write(f"You: {user_msg}\nBot: (No response)\n\n")
+    except Exception as e:
+        print("Error saving chat:", e)
 
 def process_input(user_input, history):
-    if history is None or not isinstance(history, list):
-        history = []
-    updated_history, _ = respond(user_input, history)
-    save_chat(updated_history)
-    return updated_history, "", updated_history
+    try:
+        result = respond(user_input, history)
+        save_chat(result[0])
+        return result
+    except Exception as e:
+        error_message = f"Error: {str(e)}"
+        if not isinstance(history, list):
+            history = []
+        history.append((user_input, error_message))
+        return history, "", history
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Type your message here...")
+    msg = gr.Textbox(placeholder="Type your message here...", show_label=False)
     state = gr.State([])
 
     msg.submit(process_input, inputs=[msg, state], outputs=[chatbot, msg, state])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()