Update src/ui.py

src/ui.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from src.logic import generate_answer, rebuild
 
+
 def create_ui():
     with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="MindMesh") as demo:
         gr.Markdown(
@@ -13,6 +14,7 @@ def create_ui():
         )
 
         with gr.Row(equal_height=False):
+            # Left Section — user question + buttons
             with gr.Column(scale=2):
                 q = gr.Textbox(
                     label="Your Question",
@@ -20,9 +22,14 @@ def create_ui():
                     lines=2,
                     show_label=True,
                 )
-                rebuild_btn = gr.Button("🔁 Rebuild Index", variant="secondary")
+
+                with gr.Row():
+                    generate_btn = gr.Button("⚙️ Generate Insight", variant="primary")
+                    clear_btn = gr.Button("🧹 Clear", variant="secondary")
+
                 status = gr.Markdown(value="", visible=True)
 
+            # Right Section — answer mode selector
             with gr.Column(scale=1):
                 mode = gr.Radio(
                     ["Quick Summary (Offline)", "LLM Precision (Groq Llama-3.1 70B)"],
@@ -30,8 +37,10 @@ def create_ui():
                     label="Answer Mode",
                 )
 
+        # Output section
         out = gr.Markdown()
 
+        # Example prompts
         examples = gr.Examples(
             examples=[
                 ["What can startups learn from the Industrial Revolution?"],
@@ -42,14 +51,18 @@ def create_ui():
             inputs=q
         )
 
+        # Button logic
+        generate_btn.click(fn=lambda x, m: generate_answer(x, m), inputs=[q, mode], outputs=out)
+        clear_btn.click(lambda: ("", "", ""), outputs=[q, out, status])
         q.submit(fn=lambda x, m: generate_answer(x, m), inputs=[q, mode], outputs=out)
-        rebuild_btn.click(fn=rebuild, inputs=None, outputs=status)
 
+        # Styling
         gr.HTML("""
         <style>
-        #component-0 { max-width: 900px; margin: auto; }
+        #component-0 { max-width: 950px; margin: auto; }
         textarea { font-size: 15px !important; }
         button { font-size: 14px !important; font-weight: 600; border-radius: 8px !important; }
+        .gradio-container { padding-top: 0 !important; }
         </style>
         """)