update app.py

app.py

@@ -40,7 +40,6 @@ def generate_questions(topic, difficulty, types, count):
             task="text-generation",
             model=model,
             tokenizer=tokenizer,
-            # device=0,
             max_length=200,
             temperature=0.7,
             top_p=0.9,
@@ -67,10 +66,25 @@ def generate_questions(topic, difficulty, types, count):
                 generated_text = result[0]['generated_text'].replace(formatted_prompt, "").strip()
                 parsed = parse_generated_text(generated_text)
 
-                questions.append(f"**Type**: {q_type}\n\n**Question**: {parsed['question']}\n\n**Hint**: {parsed['hint']}\n\n---")
+                print("Parsed Output:", parsed)
 
-        return "\n\n".join(questions)
+                # Safe fallback
+                if not parsed['question']:
+                    parsed['question'] = "⚠️ Could not parse question."
+                if not parsed['hint']:
+                    parsed['hint'] = "No hint available"
 
+                formatted = f"**Type**: {q_type}\n\n**Question**: {parsed['question']}\n\n**Hint**: {parsed['hint']}\n\n---"
+                questions.append(formatted)
+
+        final_output = "\n\n".join(questions)
+        print("Final Output:", final_output)
+        return final_output
+
+    except Exception as e:
+        print("Error:", e)
+        return f"❌ Something went wrong: {e}"
+    
     except Exception as e:
         print("Error:", e)
         return f"Something went wrong: {e}"
@@ -88,5 +102,5 @@ iface = gr.Interface(
     description="Enter a topic, select difficulty and question types to generate AI-powered questions."
 )
 
-iface.queue()
-iface.launch()
+if __name__ == "__main__":
+    iface.queue().launch()