Update app.py

app.py

@@ -15,6 +15,7 @@ template = """Meet Arun, your youthful and witty personal assistant!
 At 21 years old, he is full of energy and always eager to help. 
 Arun's goal is to assist you with any questions or problems you might have. 
 His enthusiasm shines through in every response, making interactions enjoyable and engaging.
+
 {chat_history}
 User: {user_message}
 Chatbot:"""
@@ -31,7 +32,8 @@ llm = ChatOpenAI(
     api_key=openai_api_key
 )
 
-# Function with per-user memory
+
+# --- Core Function ---
 def get_text_response(user_message, history, memory_state):
     if memory_state is None:
         memory_state = ConversationBufferMemory(memory_key="chat_history")
@@ -44,32 +46,33 @@ def get_text_response(user_message, history, memory_state):
     )
 
     response = llm_chain.predict(user_message=user_message)
-    return response, memory_state
+    history = history + [[user_message, response]]
+    return history, memory_state
+
 
-# Gradio Chat App with State (per-user memory)
+# --- Gradio UI ---
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Ask me anything...", show_label=False)
-    send_btn = gr.Button("Send")   # 👈 Optional send button
-    memory_state = gr.State()      # 👈 each user gets their own memory
+    msg = gr.Textbox(placeholder="Type your message...", show_label=False)
+    send_btn = gr.Button("Send")
+    memory_state = gr.State()
 
     def respond(message, history, memory_state):
-        reply, memory_state = get_text_response(message, history, memory_state)
-        history = history + [[message, reply]]
-        return history, history, memory_state, gr.Textbox.update(value="")  # clear input
+        history, memory_state = get_text_response(message, history, memory_state)
+        return history, memory_state, gr.Textbox.update(value="")  # clear input box
 
-    # Handle Enter key
+    # Enter key submit
     msg.submit(
         respond,
         [msg, chatbot, memory_state],
-        [chatbot, chatbot, memory_state, msg]
+        [chatbot, memory_state, msg]
     )
 
-    # Handle Send button
+    # Send button click
     send_btn.click(
         respond,
         [msg, chatbot, memory_state],
-        [chatbot, chatbot, memory_state, msg]
+        [chatbot, memory_state, msg]
     )
 
 if __name__ == "__main__":