Update app.py

app.py

@@ -15,17 +15,27 @@ headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 # ----------------- FEATURE FUNCTIONS -----------------
 
 def generate_quiz(text, num_questions):
-    prompt = f"Generate {num_questions} MCQ questions from this content:\n{text}"
+    prompt = f"Generate {num_questions} multiple choice questions from the following content:\n\n{text}\n\nInclude options A, B, C, D and correct answer."
+
     API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-small"
     payload = {
         "inputs": prompt,
-        "parameters": {"max_new_tokens": 256}
+        "parameters": {
+            "temperature": 0.7,
+            "max_new_tokens": 300
+        }
     }
+
     response = requests.post(API_URL, headers=headers, json=payload)
     try:
-        return response.json()[0]["generated_text"]
+        result = response.json()
+        if isinstance(result, list) and "generated_text" in result[0]:
+            return result[0]["generated_text"]
+        else:
+            return "⚠️ Model returned no valid output. Try changing the input or reduce question count."
     except Exception as e:
-        return f"⚠️ Error: {e}"
+        return f"⚠️ API Error: {str(e)}"
+
 
 
 def get_bot_response(query):