Update app.py

app.py

@@ -25,6 +25,17 @@ 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)":
+            return ans_a
+        elif final_answer[:2] == "B)":
+            return ans_b
+        elif final_answer[:2] == "C)":
+            return ans_c
+        elif final_answer[:3] == "D)":
+            return ans_d
+        else
+            return final_answer
+        
     except IndexError:
         final_answer = "Could not parse the model's answer."