Update app.py

app.py

@@ -7,12 +7,12 @@ def handle_query(query):
     for result in sanchaari(query):
         if "thought" in result:
             thoughts.append(result["thought"])
-            yield {"Thought Process": "\n".join(thoughts), "Response Text": ""}
+            yield "\n".join(thoughts), ""
         if "final_answer" in result:
             final_answer = result["final_answer"]
-            yield {"Thought Process": "\n".join(thoughts), "Response Text": final_answer}
+            yield "\n".join(thoughts), final_answer
 
-with gr.Blocks(css="footer {visibility: hidden}") as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
         # Antarjaala Sanchaari: Your Versatile AI Assistant
@@ -31,7 +31,8 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
             submit_btn = gr.Button("Ask Sanchaari", variant="primary")
         
         with gr.Column(scale=3):
-            output = gr.JSON(label="Sanchaari's Response")
+            thought_output = gr.Textbox(label="Thought Process", lines=5)
+            response_output = gr.Textbox(label="Sanchaari's Response", lines=3)
     
     gr.Markdown(
         """
@@ -52,6 +53,6 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
         """
     )
     
-    submit_btn.click(handle_query, inputs=query_input, outputs=output)
+    submit_btn.click(handle_query, inputs=query_input, outputs=[thought_output, response_output])
 
 demo.launch(share=True)