Update app.py

app.py

@@ -47,35 +47,17 @@ def chunk_text(text, chunk_size=800):
     
     return chunks
 
-def format_quiz_output(raw_output):
-    """Format the quiz output for better readability"""
-    # Split by question patterns
-    questions = re.split(r'\n(?=\d+\.|\d+\))', raw_output)
-    formatted_quiz = ""
-    
-    for i, question in enumerate(questions, 1):
-        if question.strip():
-            # Clean up the question
-            question = question.strip()
-            if not question.startswith(f"{i}."):
-                question = f"{i}. {question}"
-            
-            formatted_quiz += f"\n{question}\n"
-            formatted_quiz += "-" * 50 + "\n"
-    
-    return formatted_quiz
-
 def generate_quiz(pdf_bytes, num_questions=10):
     """Generate quiz from PDF"""
     if pdf_bytes is None:
-        return "❌ Please upload a PDF file first."
+        return "Error: Please upload a PDF file first."
     
     try:
         # Extract text from PDF
         text = extract_text_from_pdf(pdf_bytes)
         
         if not text.strip():
-            return "❌ No text found in the PDF. Please check if the PDF contains readable text."
+            return "Error: No text found in the PDF. Please check if the PDF contains readable text."
         
         # Clean the text
         text = re.sub(r'\s+', ' ', text).strip()
@@ -122,18 +104,79 @@ Text: {chunk}"""
                 continue
         
         if not all_questions:
-            return "❌ Failed to generate questions. The text might be too complex or the model encountered an error."
+            return "Error: Failed to generate questions. The text might be too complex or the model encountered an error."
         
         # Combine all questions
-        final_quiz = "\n".join(all_questions)
+        final_quiz = "\n\n".join(all_questions)
         
         # Format the output
-        formatted_quiz = f"""
-📚 QUIZ GENERATED FROM YOUR PDF
-{'='*60}
+        formatted_quiz = f"""QUIZ GENERATED FROM YOUR PDF
+============================================================
 
 {final_quiz}
 
-{'='*60}
-📝 Total Questions Generated: {len(all_questions)} sections
-💡 Review each question carefully and verify the answers."""
+============================================================
+Total Questions Generated: {len(all_questions)} sections
+Note: Review each question carefully and verify the answers.
+"""
+        
+        return formatted_quiz
+        
+    except Exception as e:
+        return f"Error processing PDF: {str(e)}\n\nPlease make sure you uploaded a valid PDF file."
+
+# Create Gradio interface
+def create_interface():
+    with gr.Blocks(title="PDF Quiz Generator") as interface:
+        gr.Markdown("""
+        # PDF to Quiz Generator
+        Upload a PDF chapter and generate multiple choice questions with answers and explanations.
+        """)
+        
+        with gr.Row():
+            with gr.Column():
+                pdf_input = gr.File(
+                    label="Upload PDF Chapter", 
+                    file_types=[".pdf"],
+                    type="binary"
+                )
+                
+                num_questions = gr.Slider(
+                    minimum=5,
+                    maximum=15,
+                    value=10,
+                    step=1,
+                    label="Number of Questions to Generate"
+                )
+                
+                generate_btn = gr.Button("Generate Quiz", variant="primary")
+            
+            with gr.Column():
+                output = gr.Textbox(
+                    label="Generated Quiz",
+                    lines=20,
+                    max_lines=30,
+                    show_copy_button=True
+                )
+        
+        generate_btn.click(
+            fn=generate_quiz,
+            inputs=[pdf_input, num_questions],
+            outputs=output
+        )
+        
+        gr.Markdown("""
+        ### Instructions:
+        1. Upload a PDF file containing the chapter or text you want to create a quiz from
+        2. Select the number of questions you want (5-15)
+        3. Click "Generate Quiz" and wait for the results
+        4. Copy the generated quiz for your use
+        
+        **Note:** The quality of questions depends on the clarity and content of your PDF text.
+        """)
+    
+    return interface
+
+if __name__ == "__main__":
+    interface = create_interface()
+    interface.launch(debug=True, share=False)