Update app.py

app.py

@@ -15,26 +15,30 @@ headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 # ----------------- FEATURE FUNCTIONS -----------------
 
 def generate_quiz(text, num_questions):
-    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."
+    prompt = (
+        f"Generate {num_questions} simple quiz questions from the following text:\n\n"
+        f"{text}\n\n"
+        f"Each question should be followed by options A, B, C, D and a correct answer."
+    )
 
     API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-small"
     payload = {
         "inputs": prompt,
-        "parameters": {
-            "temperature": 0.7,
-            "max_new_tokens": 300
-        }
+        "parameters": {"max_new_tokens": 300}
     }
 
     response = requests.post(API_URL, headers=headers, json=payload)
+    
     try:
         result = response.json()
         if isinstance(result, list) and "generated_text" in result[0]:
-            return result[0]["generated_text"]
+            output = result[0]["generated_text"].strip()
+            return output if output else "⚠️ Model returned empty text. Try reducing question count or using a smaller paragraph."
         else:
-            return "⚠️ Model returned no valid output. Try changing the input or reduce question count."
+            return f"⚠️ Model didn't return expected output. Try fewer questions or simpler content."
     except Exception as e:
-        return f"⚠️ API Error: {str(e)}"
+        return f"⚠️ API Error: {e}"
+