Update

app.py

@@ -2,9 +2,47 @@
 import gradio as gr
 import os
 import threading
+import sys
+import uuid
+import time
 from main import generate_multiple_mazes, merge_pdfs
 
-def generate_mazes_interface(num_files, width, height, progress=gr.Progress()):
+# Session management
+user_sessions = {}
+
+def create_session():
+    """Create a new session for the user"""
+    session_id = str(uuid.uuid4())[:8]  # Short session ID
+    user_sessions[session_id] = {
+        'created_at': time.time(),
+        'files_generated': 0
+    }
+    return session_id
+
+def get_session_files(session_id):
+    """Get list of PDF files for a specific session"""
+    try:
+        if os.path.exists('pdfs'):
+            pdf_files = [f for f in os.listdir('pdfs')
+                        if f.endswith('.pdf') and f.startswith(f"{session_id}_")]
+            pdf_files.sort()
+            return pdf_files if pdf_files else ["No PDF files found for this session"]
+        else:
+            return ["pdfs/ directory does not exist"]
+    except Exception as e:
+        return [f"Error: {str(e)}"]
+
+def cleanup_merged_pdf(session_id):
+    """Remove the merged PDF for a session after download"""
+    merged_pdf_path = os.path.join('pdfs', f"{session_id}_merged_mazes.pdf")
+    if os.path.exists(merged_pdf_path):
+        try:
+            os.remove(merged_pdf_path)
+            print(f"Cleaned up merged PDF: {merged_pdf_path}")
+        except Exception as e:
+            print(f"Error cleaning up merged PDF: {e}")
+
+def generate_mazes_interface(num_files, width, height, session_id, progress=gr.Progress()):
     """Interface function for generating mazes with progress tracking"""
     try:
         progress(0.1, desc="Starting maze generation...")
@@ -21,20 +59,24 @@ def generate_mazes_interface(num_files, width, height, progress=gr.Progress()):
         
         progress(0.2, desc="Generating mazes...")
         
-        # Generate the mazes
-        success_count = generate_multiple_mazes(num_files, width, height)
+        # Generate the mazes with session ID
+        success_count = generate_multiple_mazes(num_files, width, height, session_id)
         
         progress(1.0, desc="Generation complete!")
         
+        # Update session info
+        if session_id in user_sessions:
+            user_sessions[session_id]['files_generated'] += success_count
+        
         if success_count == num_files:
-            return f"Successfully generated {success_count} maze PDFs in pdfs/ directory", get_pdf_list()
+            return f"Successfully generated {success_count} maze PDFs for session {session_id}", get_session_files(session_id)
         else:
-            return f"Warning: Only {success_count} out of {num_files} mazes were generated successfully", get_pdf_list()
+            return f"Warning: Only {success_count} out of {num_files} mazes were generated successfully for session {session_id}", get_session_files(session_id)
             
     except Exception as e:
         return f"Error: {str(e)}", None
 
-def merge_pdfs_interface(progress=gr.Progress()):
+def merge_pdfs_interface(session_id, progress=gr.Progress()):
     """Interface function for merging PDFs with progress tracking"""
     try:
         progress(0.1, desc="Starting PDF merge...")
@@ -44,18 +86,19 @@ def merge_pdfs_interface(progress=gr.Progress()):
 
         progress(0.5, desc="Merging PDFs...")
 
-        success = merge_pdfs('pdfs')
+        # Merge only session-specific PDFs
+        success = merge_pdfs('pdfs', session_id)
 
         progress(1.0, desc="Merge complete!")
 
         if success:
-            merged_pdf_path = os.path.join('pdfs', 'merged_mazes.pdf')
+            merged_pdf_path = os.path.join('pdfs', f"{session_id}_merged_mazes.pdf")
             if os.path.exists(merged_pdf_path):
-                return "Successfully merged all PDFs into pdfs/merged_mazes.pdf", get_pdf_list(), merged_pdf_path
+                return f"Successfully merged session {session_id} PDFs into {session_id}_merged_mazes.pdf", get_session_files(session_id), merged_pdf_path
             else:
-                return "Merge completed but merged file not found", get_pdf_list(), None
+                return "Merge completed but merged file not found", get_session_files(session_id), None
         else:
-            return "Failed to merge PDFs", get_pdf_list(), None
+            return "Failed to merge PDFs", get_session_files(session_id), None
 
     except Exception as e:
         return f"Error: {str(e)}", None, None
@@ -82,8 +125,12 @@ def create_interface():
     os.makedirs('pdfs', exist_ok=True)
 
     with gr.Blocks(title="Maze Generator", theme=gr.themes.Soft()) as interface:
+        # Session state
+        session_id = gr.State(create_session())
+        
         gr.Markdown("# 🎯 Maze Generator")
         gr.Markdown("Generate custom maze PDFs or merge existing ones using this web interface.")
+        gr.Markdown(f"**Session ID:** {session_id.value} - Your files are isolated from other users")
         
         with gr.Tabs():
             with gr.TabItem("Generate Mazes"):
@@ -118,10 +165,10 @@ def create_interface():
                         generate_btn = gr.Button("🚀 Generate Mazes", variant="primary", size="lg")
                         
                     with gr.Column():
-                        gr.Markdown("### Generated Files")
+                        gr.Markdown("### Your Generated Files")
                         file_list = gr.List(
-                            label="PDF Files in pdfs/ directory",
-                            value=get_pdf_list()
+                            label="PDF Files for Your Session",
+                            value=get_session_files(session_id.value)
                         )
                         
                         refresh_btn = gr.Button("🔄 Refresh List", size="sm")
@@ -137,23 +184,24 @@ def create_interface():
                 # Event handlers
                 generate_btn.click(
                     fn=generate_mazes_interface,
-                    inputs=[num_files, width, height],
+                    inputs=[num_files, width, height, session_id],
                     outputs=[generate_output, file_list],
                     show_progress=True
                 )
                 
                 refresh_btn.click(
-                    fn=refresh_file_list,
+                    fn=lambda sid: get_session_files(sid),
+                    inputs=[session_id],
                     outputs=[file_list]
                 )
             
             with gr.TabItem("Merge PDFs"):
                 with gr.Row():
                     with gr.Column():
-                        gr.Markdown("### Merge Existing PDFs")
-                        gr.Markdown("Merge all PDF files in the pdfs/ directory into a single file and delete individual files.")
+                        gr.Markdown("### Merge Your PDFs")
+                        gr.Markdown("Merge all your PDF files in this session into a single file. Individual files will be deleted after merging.")
 
-                        merge_btn = gr.Button("🔗 Merge All PDFs", variant="primary", size="lg")
+                        merge_btn = gr.Button("🔗 Merge Your PDFs", variant="primary", size="lg")
 
                         merge_output = gr.Textbox(
                             label="Merge Status",
@@ -164,8 +212,8 @@ def create_interface():
                         )
 
                         merge_file_list = gr.List(
-                            label="PDF Files After Merge",
-                            value=get_pdf_list()
+                            label="Your PDF Files After Merge",
+                            value=get_session_files(session_id.value)
                         )
 
                         download_merged_btn = gr.DownloadButton(
@@ -177,24 +225,44 @@ def create_interface():
                 # Event handlers
                 merge_btn.click(
                     fn=merge_pdfs_interface,
+                    inputs=[session_id],
                     outputs=[merge_output, merge_file_list, download_merged_btn],
                     show_progress=True
                 )
+                
+                # Clean up merged PDF after download
+                download_merged_btn.click(
+                    fn=lambda sid: cleanup_merged_pdf(sid),
+                    inputs=[session_id]
+                )
         
         # Auto-refresh file list every few seconds
         interface.load(
-            fn=lambda: (get_pdf_list(), get_pdf_list()),
+            fn=lambda sid: (get_session_files(sid), get_session_files(sid)),
+            inputs=[session_id],
             outputs=[file_list, merge_file_list]
         )
     
     return interface
 
 if __name__ == "__main__":
+    print("🚀 Starting Maze Generator Application...")
+    print("📁 Creating pdfs directory if it doesn't exist...")
+    
     # Create and launch the interface
     interface = create_interface()
+    
+    print("🌐 Launching Gradio interface...")
+    print("📍 Server will be available at: http://0.0.0.0:7861")
+    print("📍 Local access: http://localhost:7861")
+    print("⏳ Waiting for server to start...")
+    
     interface.launch(
         server_name="0.0.0.0",
         server_port=7860,
         share=False,
         show_error=True
-    )
+    )
+    
+    print("✅ Maze Generator is ready and running!")
+    print("🎯 Open your browser and navigate to http://localhost:7861")

main.py

@@ -182,7 +182,7 @@ def extract_first_page(input_pdf_path, output_pdf_path):
         print(f"Error extracting first page: {e}")
         return False
 
-def generate_multiple_mazes(num_files, width, height):
+def generate_multiple_mazes(num_files, width, height, session_id=None):
     """Generate multiple maze PDFs with specified dimensions"""
     # Create pdfs directory if it doesn't exist
     os.makedirs('pdfs', exist_ok=True)
@@ -198,9 +198,10 @@ def generate_multiple_mazes(num_files, width, height):
         # Download the full PDF
         temp_filename = f"temp_maze_{i+1}.pdf"
         if download_pdf(maze_html, maze_cookies, temp_filename):
-            # Extract only the first page to pdfs directory
+            # Extract only the first page to pdfs directory with session prefix
             # Start with i+1, but get_unique_filename will find the next available number
-            final_filename = os.path.join('pdfs', f"maze_{width}x{height}_{i+1}.pdf")
+            session_prefix = f"{session_id}_" if session_id else ""
+            final_filename = os.path.join('pdfs', f"{session_prefix}maze_{width}x{height}_{i+1}.pdf")
             result_path = extract_first_page(temp_filename, final_filename)
             
             if result_path:
@@ -217,11 +218,20 @@ def generate_multiple_mazes(num_files, width, height):
     print(f"\nSuccessfully generated {success_count}/{num_files} maze PDFs in pdfs/ directory")
     return success_count
 
-def merge_pdfs(pdf_directory, output_filename="merged_mazes.pdf"):
+def merge_pdfs(pdf_directory, session_id=None, output_filename=None):
     """Merge all PDFs in directory into a single file and delete individual files"""
     try:
         # Get all PDF files in directory
-        pdf_files = [f for f in os.listdir(pdf_directory) if f.endswith('.pdf')]
+        if session_id:
+            # Only merge files with the session prefix
+            pdf_files = [f for f in os.listdir(pdf_directory)
+                        if f.endswith('.pdf') and f.startswith(f"{session_id}_") and not f.endswith("_merged_mazes.pdf")]
+            output_filename = output_filename or f"{session_id}_merged_mazes.pdf"
+        else:
+            # Merge all PDF files (backward compatibility)
+            pdf_files = [f for f in os.listdir(pdf_directory) if f.endswith('.pdf')]
+            output_filename = output_filename or "merged_mazes.pdf"
+        
         pdf_files.sort()  # Sort files by name
         
         if not pdf_files:
@@ -284,7 +294,7 @@ def main():
     if args.merge:
         print("Merge mode: Merging all existing PDFs in pdfs/ directory")
         if os.path.exists('pdfs'):
-            success = merge_pdfs('pdfs')
+            success = merge_pdfs('pdfs', session_id=None)
             return 0 if success else 1
         else:
             print("Error: pdfs/ directory does not exist")