Update app.py

app.py

@@ -36,8 +36,19 @@ def generate_mcqs_on_topic(text, topic, num_mcqs=5):
 
     return mcqs
 
+def extract_options_and_correct_answer(api_response):
+    if 'choices' in api_response:
+        message = api_response['choices'][0]['message']
+        content = message.get('content', "Unable to generate a question..")
+        options = message.get('options', [])
+        correct_answer = message.get('correct_answer', "Unknown")
+
+        return content, options, correct_answer
+
+    return "Unexpected API response format.", [], "Unknown"
+
 def generate_question_with_chatgpt(context, topic):
-    client=OpenAI()
+    client = OpenAI()
     # Initializing the default value
     generated_question = {
         'content': "Unable to generate a question..",
@@ -48,26 +59,17 @@ def generate_question_with_chatgpt(context, topic):
     response = client.chat.completions.create(
         model="gpt-3.5-turbo",
         max_tokens=1024,
-        temperature = 0.7,
-        messages = [
+        temperature=0.7,
+        messages=[
             {"role": "system", "content": "You are a helpful assistant."},
             {"role": "user", "content": f"What is the question on {topic} for the following? {context}"},
         ]
-
     )
-    result=response.json()
+    result = response.json()
 
     print("API Response:", result)  # Add this line for debugging
 
-    if 'choices' in result:
-        # Extract the generated question, options, and correct answer from the response
-        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")
-        }
-    else:
-        print("Unexpected API response format.")
+    generated_question['content'], generated_question['options'], generated_question['correct_answer'] = extract_options_and_correct_answer(result)
 
     return generated_question
 
@@ -102,6 +104,7 @@ if __name__ == "__main__":
 
 
 
+