Update app.py

app.py

@@ -1,15 +1,3 @@
-
-import gradio as gr
-import pandas as pd
-from mcq_generator import ImprovedMCQGenerator, is_suitable_for_students
-import io
-import tempfile
-import os
-
-
-# Load MCQ generator once
-mcq_generator = ImprovedMCQGenerator()
-
 def generate_mcqs_ui(paragraph, num_questions):
     if not paragraph.strip():
         return None, None, "⚠️ Please enter a valid paragraph."
@@ -32,7 +20,7 @@ def generate_mcqs_ui(paragraph, num_questions):
             question_html += "</div>"
             pretty_mcqs.append(question_html)
 
-        # Prepare download files
+        # Prepare text output and CSV data
         txt_output = ""
         csv_data = []
 
@@ -51,42 +39,21 @@ def generate_mcqs_ui(paragraph, num_questions):
                 'Answer': chr(65+mcq['answer_index'])
             })
 
-        # ✅ Create real temporary files
-        txt_temp = tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode='w', encoding='utf-8')
-        txt_temp.write(txt_output)
-        txt_temp.close()
+        # Create in-memory text file
+        txt_buffer = io.StringIO()
+        txt_buffer.write(txt_output)
+        txt_buffer.seek(0)
+
+        # Create in-memory CSV file
+        csv_buffer = io.StringIO()
+        pd.DataFrame(csv_data).to_csv(csv_buffer, index=False)
+        csv_buffer.seek(0)
 
-        csv_temp = tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode='w', encoding='utf-8')
-        pd.DataFrame(csv_data).to_csv(csv_temp.name, index=False)
-        csv_temp.close()
+        # Return Gradio-compatible downloadable files
+        txt_file = gr.File(value=txt_buffer, file_name="mcqs.txt", file_type="text/plain")
+        csv_file = gr.File(value=csv_buffer, file_name="mcqs.csv", file_type="text/csv")
 
-        return pretty_mcqs, [txt_temp.name, csv_temp.name], "✅ MCQs generated successfully!"
+        return pretty_mcqs, [txt_file, csv_file], "✅ MCQs generated successfully!"
 
     except Exception as e:
         return None, None, f"❌ Error generating MCQs: {str(e)}"
-
-# Gradio Interface
-with gr.Blocks(theme=gr.themes.Default()) as demo:
-    gr.Markdown("<h1 style='text-align:center;'>📚 Smart MCQ Generator</h1>")
-    with gr.Row():
-        paragraph_input = gr.Textbox(lines=8, label="Enter Paragraph for MCQs", placeholder="Paste your study material here...")
-    with gr.Row():
-        num_questions_slider = gr.Slider(1, 10, step=1, value=5, label="Number of Questions")
-    with gr.Row():
-        generate_btn = gr.Button("🚀 Generate MCQs")
-    status = gr.Textbox(label="Status", interactive=False)
-
-    with gr.Row():
-        mcq_output = gr.HTML()
-
-    with gr.Row():
-        download_output = gr.File(label="Download MCQs (TXT/CSV)")
-
-    generate_btn.click(
-        fn=generate_mcqs_ui,
-        inputs=[paragraph_input, num_questions_slider],
-        outputs=[mcq_output, download_output, status]
-    )
-
-demo.launch(ssr_mode=False)
-