Update app.py

app.py

@@ -119,10 +119,10 @@ print("Model loaded!")
 # Create Gradio interface with chat
 demo = gr.Interface(
     fn=chat_response,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your message here..."),
-    outputs=gr.Textbox(),
+    inputs=[gr.Textbox(label="Message"), gr.State()],  # Add State for chat history
+    outputs=gr.Textbox(label="Response"),  # Single response output
     title="Admissions Agent Assistant",
-    description="Chat with an AI-powered admissions coordinator. The agent will maintain context of your conversation.",
+    description="Chat with an AI-powered admissions coordinator.",
     examples=[
         "I need help with addiction treatment",
         "What insurance do you accept?",
@@ -131,5 +131,6 @@ demo = gr.Interface(
     ]
 )
 
+
 if __name__ == "__main__":
     demo.launch(share=True) # Remove s