Update app.py

app.py

@@ -25,7 +25,11 @@ prompt = PromptTemplate(
     template=template
 )
 
-# LLM with API Key
+# Memory
+def new_memory():
+    return ConversationBufferMemory(memory_key="chat_history")
+
+# LLM
 llm = ChatOpenAI(
     temperature=0.5,
     model="gpt-4o-mini",
@@ -33,47 +37,32 @@ llm = ChatOpenAI(
 )
 
 
-# --- Core Function ---
-def get_text_response(user_message, history, memory_state):
-    if memory_state is None:
-        memory_state = ConversationBufferMemory(memory_key="chat_history")
+# --- Core Chat Function ---
+def respond(user_message, history, memory_state):
+    if memory_state is None:  # create new memory for each session
+        memory_state = new_memory()
 
-    llm_chain = LLMChain(
+    chain = LLMChain(
         llm=llm,
         prompt=prompt,
-        verbose=True,
-        memory=memory_state
+        memory=memory_state,
+        verbose=False
     )
 
-    response = llm_chain.predict(user_message=user_message)
+    response = chain.predict(user_message=user_message)
     history = history + [[user_message, response]]
-    return history, memory_state
+    return history, memory_state, ""   # last "" clears textbox
 
 
 # --- Gradio UI ---
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
+    chatbot = gr.Chatbot(label="Arun AI Assistant")
     msg = gr.Textbox(placeholder="Type your message...", show_label=False)
-    send_btn = gr.Button("Send")
+    send = gr.Button("Send")
     memory_state = gr.State()
 
-    def respond(message, history, memory_state):
-        history, memory_state = get_text_response(message, history, memory_state)
-        return history, memory_state, gr.Textbox.update(value="")  # clear input box
-
-    # Enter key submit
-    msg.submit(
-        respond,
-        [msg, chatbot, memory_state],
-        [chatbot, memory_state, msg]
-    )
-
-    # Send button click
-    send_btn.click(
-        respond,
-        [msg, chatbot, memory_state],
-        [chatbot, memory_state, msg]
-    )
+    msg.submit(respond, [msg, chatbot, memory_state], [chatbot, memory_state, msg])
+    send.click(respond, [msg, chatbot, memory_state], [chatbot, memory_state, msg])
 
 if __name__ == "__main__":
     demo.launch(share=True)