Update app.py

app.py

@@ -42,7 +42,11 @@ def generate_question_with_chatgpt(context):
     }
 
     # Initializing the default value
-    generated_question = "Unable to generate a question.."
+    generated_question = {
+        'content': "Unable to generate a question..",
+        'options': [],  # assuming options is a list
+        'correct_answer': "Unknown"
+    }
 
     data = {
         "model": "gpt-3.5-turbo",
@@ -57,16 +61,12 @@ def generate_question_with_chatgpt(context):
 
     if 'choices' in result: 
         # Extract the generated question from the response
-        generated_question = result["choices"][0]["message"]["content"]
-
-        # Decode the byte string to Unicode string
-        generated_question = generated_question.decode("utf-8")
+        generated_question = {
+            'content': result["choices"][0]["message"]["content"],
+            'options': result["choices"][0]["message"].get("options", []),
+            'correct_answer': result["choices"][0]["message"].get("correct_answer", "Unknown")
+        }
 
-         # Post-process the question if needed
-        # Example: Ensure the question ends with a question mark
-        if not generated_question.endswith('?'):
-            generated_question += '?'
-            
     return generated_question
 
 def main(): 
@@ -88,7 +88,7 @@ def main():
             # Display the generated Questions
             st.success(f"Generated {num_mcqs} Questions:")
             for i, generated_question in enumerate(mcqs, start=1):
-                st.write(f"\nQuestion {i}: {generated_question}")
+                st.write(f"\nQuestion {i}: {generated_question['content']}")
                 st.write(f"Options: {', '.join(generated_question['options'])}")
                 st.write(f"Correct Answer: {generated_question['correct_answer']}")
         else:
@@ -97,3 +97,4 @@ def main():
 if __name__ == "__main__":
     main()
 
+