Update app.py

app.py

@@ -24,6 +24,9 @@ prompt = PromptTemplate(
     template=template
 )
 
+# Memory (Conversation history)
+memory = ConversationBufferMemory(memory_key="chat_history")
+
 # LLM with API Key
 llm = ChatOpenAI(
     temperature=0.5,
@@ -31,32 +34,27 @@ llm = ChatOpenAI(
     api_key=openai_api_key
 )
 
-# ✅ Chat Function (3 args required!)
-def get_text_response(user_message, history, state):
-    # create memory if none exists for this session
-    if state is None:
-        state = ConversationBufferMemory(memory_key="chat_history")
-
-    # Chain with per-user memory
-    llm_chain = LLMChain(
-        llm=llm,
-        prompt=prompt,
-        verbose=True,
-        memory=state
-    )
+# LLM Chain
+llm_chain = LLMChain(
+    llm=llm,
+    prompt=prompt,
+    verbose=True,
+    memory=memory
+)
 
+# Chat Function
+def get_text_response(user_message, history):
     response = llm_chain.predict(user_message=user_message)
-    return response, state
+    return response
 
-# ✅ Gradio Chat App
+# Gradio Chat App
 demo = gr.ChatInterface(
     fn=get_text_response,
     examples=[
-        ["How are you doing?", None],
-        ["What are your interests?", None],
-        ["Which places do you like to visit?", None]
-    ],
-    additional_inputs=[gr.State()]
+        "How are you doing?",
+        "What are your interests?",
+        "Which places do you like to visit?"
+    ]
 )
 
 if __name__ == "__main__":