Update app.py

app.py

@@ -77,18 +77,20 @@ def generate_question_with_chatgpt(context, topic):
         choice = result['choices'][0]
         if 'message' in choice and isinstance(choice['message'], dict):
             message = choice['message']
-            content = message.get('content', "Unable to generate a question..")
-            options = message.get('options', [])
-            correct_answer = message.get('correct_answer', "Unknown")
+            if 'content' in message:
+                content = message['content']
+                options = message.get('options', [])
+                correct_answer = message.get('correct_answer', "Unknown")
 
-            generated_question['content'] = content
-            generated_question['options'] = options
-            generated_question['correct_answer'] = correct_answer
+                generated_question['content'] = content
+                generated_question['options'] = options
+                generated_question['correct_answer'] = correct_answer
 
     return generated_question
 
 
 
+
 def main(): 
     # Title of the Application
     st.header("🤖CB Quiz Generator🧠", divider='rainbow')