Update app.py

app.py

@@ -25,13 +25,13 @@ def get_answer(question, ans_a, ans_b, ans_c, ans_d):
     # Parse the final answer
     try:
         final_answer = answer_text.split("Answer:")[1].strip().split('\n')[0]
-        if final_answer[:2] == "A)":
+        if final_answer.startswith("A)"):
             return ans_a
-        elif final_answer[:2] == "B)":
+        elif final_answer.startswith("B)"):
             return ans_b
-        elif final_answer[:2] == "C)":
+        elif final_answer.startswith("C)"):
             return ans_c
-        elif final_answer[:3] == "D)":
+        elif final_answer.startswith("D)"):
             return ans_d
         else:
             return final_answer
@@ -42,7 +42,11 @@ def get_answer(question, ans_a, ans_b, ans_c, ans_d):
     return final_answer
 
 # --- Gradio Interface ---
-with gr.Blocks() as demo:
+
+# Define a soft theme with a green primary color for a more modern look
+theme = gr.themes.Soft(primary_hue="green")
+
+with gr.Blocks(theme=theme) as demo:
     gr.Markdown("# EdNA: MCQ Answering App")
     gr.Markdown("Enter a multiple-choice question and four possible answers, then click 'Get Answer' to see the model's prediction.")
 
@@ -53,8 +57,8 @@ with gr.Blocks() as demo:
         answer_c_input = gr.Textbox(label="Answer C", placeholder="Enter answer C")
         answer_d_input = gr.Textbox(label="Answer D", placeholder="Enter answer D")
 
-    get_answer_button = gr.Button("Get Answer")
-    final_answer_output = gr.Textbox(label="The Model's Answer")
+    get_answer_button = gr.Button("Get Answer", variant="primary")
+    final_answer_output = gr.Textbox(label="The Model's Answer", interactive=False)
 
     get_answer_button.click(
         fn=get_answer,