Improvised UI

app.py

@@ -52,7 +52,12 @@ tools = [
     Tool(
         name="Legal-Library",
         func=retrieve_from_db,
-        description="Retrieves relevant legal information from pre-loaded documents"
+        description=(
+        "Searches a legal document database including the Indian Penal Code, "
+        "Constitution of India, and Transfer of Property Act to retrieve accurate, "
+        "contextual, and relevant legal information. Use this tool for queries "
+        "related to specific laws, sections, or provisions in these documents."
+        )
     )
 ]
 
@@ -69,7 +74,6 @@ agent = initialize_agent(
 
 def chatbot(input_text, chat_history):
     try:    
-        
         response = agent.run(input_text)
 
         if response == "N/A":
@@ -78,7 +82,6 @@ def chatbot(input_text, chat_history):
         memory.save_context({"user": input_text}, {"assistant": response})
 
         chat_history.append([input_text, response])
-    
         
         return chat_history
     
@@ -92,18 +95,28 @@ def chatbot(input_text, chat_history):
 def clear_input():
     return gr.update(value="")
 
+def clear_chat():
+    return [],""
+
 def gradio_interface():
     with gr.Blocks() as demo:
-        gr.Markdown("# Legal Query Chatbot")
+        gr.Markdown("""
+            <div style="text-align: center;">
+                <h1>Legal Query Chatbot</h1>
+            </div>
+        """)
 
         with gr.Column():
             chatbot_ui = gr.Chatbot()
-            user_input = gr.Textbox(show_label=True, placeholder="Enter your INDIAN PENAL CODE, TRANSFER OF PROPERTY, CONSTITUTION OF INDIA query here...")
-            submit_button = gr.Button("Submit")
+            user_input = gr.Textbox(placeholder="Ask your legal questions here, such as IPC sections, property laws, constitution articles.")
+            submit_button = gr.Button("Submit", elem_classes="gr-button", variant="primary")
+            clear_chat_button = gr.Button("Clear Chat", variant="secondary")
             submit_button.click(fn=chatbot, inputs=[user_input, chatbot_ui], outputs=chatbot_ui)
             submit_button.click(fn=clear_input, inputs=None, outputs=user_input)
             user_input.submit(fn=chatbot, inputs=[user_input, chatbot_ui], outputs=chatbot_ui)
             user_input.submit(fn=clear_input, inputs=None, outputs=user_input)
+            clear_chat_button.click(fn=clear_chat, inputs=None, outputs=[chatbot_ui, user_input])
+
 
     return demo