Improvised UI

app.py

@@ -83,18 +83,15 @@ def chatbot(input_text, chat_history):
 
         chat_history.append([input_text, response])
         
-        return chat_history
+        return chat_history, ""
     
     except OutputParserException as e:
         error_message = exceptionMsg
         chat_history.append([error_message, input_text])
         print("Error:", str(e))
 
-        return chat_history
+        return chat_history, ""
     
-def clear_input():
-    return gr.update(value="")
-
 def clear_chat():
     return [],""
 
@@ -111,10 +108,8 @@ def gradio_interface():
             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)
+            submit_button.click(fn=chatbot, inputs=[user_input, chatbot_ui], outputs=[chatbot_ui, user_input])
+            user_input.submit(fn=chatbot, inputs=[user_input, chatbot_ui], outputs=[chatbot_ui, user_input])
             clear_chat_button.click(fn=clear_chat, inputs=None, outputs=[chatbot_ui, user_input])