Update app.py

app.py

@@ -34,41 +34,73 @@ def qa_system(method, question):
 # -------------------------------
 # Gradio UI
 # -------------------------------
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+with gr.Blocks(css="""
+    .radio-vertical .wrap {
+        flex-direction: column !important;
+    }
+    .radio-vertical .wrap > label {
+        margin-bottom: 8px !important;
+        margin-right: 0 !important;
+    }
+    .small-btn {
+        max-width: fit-content !important;
+        width: auto !important;
+    }
+    .small-btn button {
+        width: auto !important; 
+        min-width: unset !important; 
+        padding: 8px 16px !important; 
+        font-size: 16px !important;
+        white-space: nowrap !important;
+        max-width: fit-content !important;
+    }
+""") as demo:
     gr.Markdown(
         """
         # 📊 Comparative Financial QA System  
-        Compare **Retrieval-Augmented Generation (RAG)**  
-        and **Fine-Tuned TinyLLaMA LoRA** models for Microsoft's financial Q&A.
+        An implementation comparing **Retrieval-Augmented Generation (RAG)** and a **Fine-Tuned on LoRA and Replay-Based Learning** GPT 2 model for answering questions on financial reports.
         """
     )
 
-    with gr.Row():
-        method = gr.Radio(
-            choices=["Retrieval-Augmented Generation (RAG)", "Fine-Tuned Model"],
-            label="Choose QA Method:",
-            value="Fine-Tuned Model"
-        )
+    # Radio buttons displayed vertically
+    method = gr.Radio(
+        choices=["Retrieval-Augmented Generation (RAG)", "Fine-Tuned Model"],
+        label="Choose QA Method:",
+        value="Fine-Tuned Model",
+        interactive=True,
+        elem_classes="radio-vertical"
+    )
 
+    # Question input
     question = gr.Textbox(
         label="Ask a question about Microsoft's 2022-2023 financials:",
         placeholder="e.g., What was the total revenue in 2023?"
     )
 
-    submit_btn = gr.Button("Get Answer")
+    # Get Answer button — auto-sized
+    submit_btn = gr.Button("Get Answer", elem_classes="small-btn")
 
-    # Output section
-    method_output = gr.Markdown()
-    confidence_output = gr.Number(label="Model Confidence")
-    response_time_output = gr.Textbox(label="Response Time")
-    answer_output = gr.Markdown(label="Answer")
+    # Output section - initially hidden
+    with gr.Group(visible=False) as output_section:
+        method_output = gr.Markdown()
+        confidence_output = gr.Number(label="Model Confidence")
+        response_time_output = gr.Textbox(label="Response Time")
+        answer_output = gr.Markdown(label="Answer")
 
+    # Button click handler
+    def handle_submit(method_val, question_val):
+        # Show output section and get results
+        results = qa_system(method_val, question_val)
+        return [gr.Group(visible=True)] + list(results)
+    
     submit_btn.click(
-        qa_system,
+        handle_submit,
         inputs=[method, question],
-        outputs=[method_output, confidence_output, response_time_output, answer_output]
+        outputs=[output_section, method_output, confidence_output, response_time_output, answer_output]
     )
 
 # -------------------------------
 # Launch for Hugging Face Spaces
-demo.launch()
+# -------------------------------
+if __name__ == "__main__":
+    demo.launch()