Update app.py

app.py

@@ -1,19 +1,25 @@
 import gradio as gr
 from question_generator import generate_microbiology_question
+import logging
 
 current_question = None
 
 def generate_question():
     global current_question
-    current_question = generate_microbiology_question()
-    return (
-        current_question['question'],
-        current_question['options']['A'],
-        current_question['options']['B'],
-        current_question['options']['C'],
-        current_question['options']['D'],
-        current_question['options']['E'],
-    )
+    try:
+        current_question = generate_microbiology_question()
+        return (
+            current_question['question'],
+            current_question['options']['A'],
+            current_question['options']['B'],
+            current_question['options']['C'],
+            current_question['options']['D'],
+            current_question['options']['E'],
+            ""  # Clear any previous error messages
+        )
+    except Exception as e:
+        logging.error(f"Error generating question: {e}")
+        return ("Error generating question. Please try again.", "", "", "", "", "", str(e))
 
 def check_answer(choice):
     if current_question is None:
@@ -46,9 +52,12 @@ with gr.Blocks() as demo:
     with gr.Row():
         result = gr.Textbox(label="Result", lines=10)
     
+    with gr.Row():
+        error_message = gr.Textbox(label="Error", lines=2)
+    
     generate_btn.click(
         generate_question,
-        outputs=[question_text, option_a, option_b, option_c, option_d, option_e]
+        outputs=[question_text, option_a, option_b, option_c, option_d, option_e, error_message]
     )
     
     option_a.click(lambda: check_answer("A"), outputs=result)