Create app.py

app.py

@@ -0,0 +1,228 @@
+"""
+Enhanced Gradio application with more dynamic question configuration
+"""
+
+import gradio as gr
+import os
+import tempfile
+from multi_agent import MultiAgentSystem
+from docx_builder import generate_all_documents
+
+# Initialize the multi-agent system
+agent_system = MultiAgentSystem()
+
+def generate_exam_package(subject, stream, part_a_count, part_b_count, part_c_count, 
+                         part_a_marks, part_b_marks, part_c_marks, syllabus_file, reference_file):
+    """Enhanced main function with dynamic marks configuration"""
+    
+    try:
+        # Convert counts and marks to integers
+        part_a_count = int(part_a_count)
+        part_b_count = int(part_b_count) 
+        part_c_count = int(part_c_count)
+        part_a_marks = int(part_a_marks)
+        part_b_marks = int(part_b_marks)
+        part_c_marks = int(part_c_marks)
+        
+        # Generate exam package with dynamic marks
+        final_output, status = agent_system.generate_exam_package(
+            subject=subject,
+            stream=stream,
+            part_a_count=part_a_count,
+            part_b_count=part_b_count,
+            part_c_count=part_c_count,
+            part_a_marks=part_a_marks,
+            part_b_marks=part_b_marks, 
+            part_c_marks=part_c_marks,
+            syllabus_file=syllabus_file,
+            reference_file=reference_file
+        )
+        
+        if status != "Success":
+            return None, None, None, status
+        
+        # Create temporary directory for outputs
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Generate DOCX files
+            files = generate_all_documents(final_output, subject, stream, temp_dir)
+            
+            # Return file paths
+            return files["question_paper"], files["answer_key"], files["obe_summary"], "Generation completed successfully!"
+    
+    except Exception as e:
+        error_msg = f"Error: {str(e)}"
+        print(error_msg)
+        return None, None, None, error_msg
+
+def update_progress_ui():
+    """Update progress in UI"""
+    progress = agent_system.get_progress()
+    return f"{progress['stage'].upper()}: {progress['message']} ({progress['percentage']}%)"
+
+def calculate_total_marks(part_a_count, part_b_count, part_c_count, part_a_marks, part_b_marks, part_c_marks):
+    """Calculate and display total marks in real-time"""
+    try:
+        total = (int(part_a_count) * int(part_a_marks)) + (int(part_b_count) * int(part_b_marks)) + (int(part_c_count) * int(part_c_marks))
+        return f"📊 Total Marks: {total}"
+    except:
+        return "📊 Total Marks: 0"
+
+# Enhanced Gradio interface
+with gr.Blocks(title="AI Exam Generator", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🚀 AI-Powered Question Paper & Answer Generator")
+    gr.Markdown("Generate complete exam packages with OBE compliance using multi-agent AI system")
+    
+    with gr.Row():
+        with gr.Column():
+            # Input section
+            subject_name = gr.Textbox(
+                label="Subject Name",
+                placeholder="e.g., Data Structures, Thermodynamics, Digital Electronics",
+                value="Data Structures"
+            )
+            
+            stream = gr.Dropdown(
+                choices=["CSE", "Non-CSE"],
+                label="Stream",
+                value="CSE"
+            )
+            
+            # Enhanced question configuration with marks
+            with gr.Group():
+                gr.Markdown("### Part A Configuration")
+                with gr.Row():
+                    part_a_count = gr.Number(
+                        label="Number of Questions",
+                        value=5,
+                        minimum=1,
+                        maximum=50,
+                        step=1
+                    )
+                    part_a_marks = gr.Number(
+                        label="Marks per Question",
+                        value=2,
+                        minimum=1, 
+                        maximum=10,
+                        step=1
+                    )
+            
+            with gr.Group():
+                gr.Markdown("### Part B Configuration")
+                with gr.Row():
+                    part_b_count = gr.Number(
+                        label="Number of Questions",
+                        value=5,
+                        minimum=1,
+                        maximum=20,
+                        step=1
+                    )
+                    part_b_marks = gr.Number(
+                        label="Marks per Question", 
+                        value=13,
+                        minimum=5,
+                        maximum=20,
+                        step=1
+                    )
+            
+            with gr.Group():
+                gr.Markdown("### Part C Configuration")
+                with gr.Row():
+                    part_c_count = gr.Number(
+                        label="Number of Questions",
+                        value=1,
+                        minimum=1,
+                        maximum=5,
+                        step=1
+                    )
+                    part_c_marks = gr.Number(
+                        label="Marks per Question",
+                        value=14,
+                        minimum=10,
+                        maximum=30,
+                        step=1
+                    )
+            
+            # Total marks display
+            total_marks_display = gr.Textbox(
+                label="Marks Summary",
+                value="📊 Total Marks: 0",
+                interactive=False
+            )
+            
+            syllabus_pdf = gr.File(
+                label="Syllabus PDF",
+                file_types=[".pdf"],
+                type="filepath"
+            )
+            
+            reference_qp = gr.File(
+                label="Reference Question Paper PDF (Optional)",
+                file_types=[".pdf"],
+                type="filepath"
+            )
+            
+            generate_btn = gr.Button("Generate Exam Package", variant="primary", size="lg")
+        
+        with gr.Column():
+            # Output section
+            status_display = gr.Textbox(
+                label="Generation Status",
+                value="Ready to generate",
+                interactive=False
+            )
+            
+            with gr.Row():
+                qp_download = gr.File(
+                    label="Question Paper (.docx)",
+                    file_types=[".docx"],
+                    interactive=False
+                )
+                
+                answers_download = gr.File(
+                    label="Answer Key (.docx)", 
+                    file_types=[".docx"],
+                    interactive=False
+                )
+                
+                obe_download = gr.File(
+                    label="OBE Summary (.docx)",
+                    file_types=[".docx"],
+                    interactive=False
+                )
+    
+    # Progress section
+    progress_text = gr.Textbox(
+        label="Real-time Progress",
+        value="System ready",
+        interactive=False,
+        every=2
+    )
+    
+    # Event handlers for real-time calculations
+    inputs_for_calculation = [part_a_count, part_b_count, part_c_count, part_a_marks, part_b_marks, part_c_marks]
+    
+    for input_component in inputs_for_calculation:
+        input_component.change(
+            fn=calculate_total_marks,
+            inputs=inputs_for_calculation,
+            outputs=[total_marks_display]
+        )
+    
+    # Main generation handler
+    generate_btn.click(
+        fn=generate_exam_package,
+        inputs=[subject_name, stream, part_a_count, part_b_count, part_c_count, 
+                part_a_marks, part_b_marks, part_c_marks, syllabus_pdf, reference_qp],
+        outputs=[qp_download, answers_download, obe_download, status_display]
+    )
+    
+    # Progress updates
+    progress_text.change(
+        fn=update_progress_ui,
+        inputs=[],
+        outputs=[progress_text]
+    )
+
+if __name__ == "__main__":
+    demo.queue(concurrency_count=1)
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)