Update app.py

app.py

@@ -1,26 +1,6 @@
 import gradio as gr
 import os
 from datetime import datetime
-import json
-from http.cookies import SimpleCookie
-
-def load_user_consent():
-    """Load user consent status from cookies"""
-    try:
-        with open('user_consent.json', 'r') as f:
-            return json.load(f)
-    except:
-        return {}
-
-def save_user_consent(user_id, consent):
-    """Save user consent status to cookies"""
-    consents = load_user_consent()
-    consents[user_id] = {
-        'nsfw_consent': consent,
-        'timestamp': datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    }
-    with open('user_consent.json', 'w') as f:
-        json.dump(consents, f)
 
 def generate_image(prompt, style, num_images=1, width=512, height=512):
     """
@@ -31,9 +11,11 @@ def generate_image(prompt, style, num_images=1, width=512, height=512):
         model = gr.load("models/LAYEK-143/FLUX_V0")
         images = []
         for i in range(num_images):
+            # Generate unique filename with timestamp
             timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
             filename = f"generated_{timestamp}_{i}.png"
             
+            # Generate the image
             result = model.predict(
                 prompt=prompt,
                 style=style,
@@ -41,6 +23,7 @@ def generate_image(prompt, style, num_images=1, width=512, height=512):
                 height=height
             )
             
+            # Save the image
             os.makedirs("generated_images", exist_ok=True)
             save_path = os.path.join("generated_images", filename)
             result.save(save_path)
@@ -53,6 +36,7 @@ def generate_image(prompt, style, num_images=1, width=512, height=512):
 def create_interface():
     """Create and configure the Gradio interface"""
     
+    # Define available style options
     STYLES = [
         "Realistic",
         "Artistic",
@@ -60,143 +44,69 @@ def create_interface():
         "Cartoon",
         "Sketch"
     ]
-
-    # State variables for consent management
-    has_consented = gr.State(False)
-    user_id = gr.State(datetime.now().strftime("%Y%m%d%H%M%S"))  # Simple user identification
     
-    with gr.Blocks(title="Advanced Image Generator") as interface:
-        # Content Warning Modal
-        with gr.Box(visible=True) as warning_modal:
-            gr.Markdown("""
-            # ⚠️ Content Warning
-            
-            This image generation tool can potentially create sensitive or NSFW content. 
-            By proceeding, you acknowledge that:
-            
-            - You are 18 years or older
-            - You understand that generated content may be inappropriate
-            - You accept responsibility for the content you generate
-            - You will not use this tool to create harmful or illegal content
-            
-            Do you accept these terms?
-            """)
-            
-            with gr.Row():
-                accept_btn = gr.Button("Accept", variant="primary")
-                decline_btn = gr.Button("Decline", variant="secondary")
+    # Create the interface
+    with gr.Blocks(title="Image Generator") as interface:
+        gr.Markdown("# 🎨 Image Generator")
         
-        # Main Interface (initially hidden)
-        with gr.Box(visible=False) as main_interface:
-            gr.Markdown("# 🎨 Advanced Image Generator")
-            
-            # NSFW Warning Banner
-            with gr.Box(elem_classes="warning-banner"):
-                gr.Markdown("""
-                ⚠️ **NSFW Warning**: This tool can generate sensitive content. 
-                Please use responsibly and ensure compliance with all applicable laws and guidelines.
-                """)
-            
-            with gr.Row():
-                with gr.Column():
-                    prompt = gr.Textbox(
-                        label="Prompt",
-                        placeholder="Describe the image you want to generate...",
-                        lines=3
+        with gr.Row():
+            with gr.Column():
+                # Input components
+                prompt = gr.Textbox(
+                    label="Prompt",
+                    placeholder="Describe the image you want to generate...",
+                    lines=3
+                )
+                
+                style = gr.Dropdown(
+                    choices=STYLES,
+                    label="Style",
+                    value="Realistic"
+                )
+                
+                with gr.Row():
+                    num_images = gr.Slider(
+                        minimum=1,
+                        maximum=4,
+                        value=1,
+                        step=1,
+                        label="Number of Images"
                     )
                     
-                    style = gr.Dropdown(
-                        choices=STYLES,
-                        label="Style",
-                        value="Realistic"
+                with gr.Row():
+                    width = gr.Slider(
+                        minimum=256,
+                        maximum=1024,
+                        value=512,
+                        step=64,
+                        label="Width"
+                    )
+                    height = gr.Slider(
+                        minimum=256,
+                        maximum=1024,
+                        value=512,
+                        step=64,
+                        label="Height"
                     )
-                    
-                    with gr.Row():
-                        num_images = gr.Slider(
-                            minimum=1,
-                            maximum=4,
-                            value=1,
-                            step=1,
-                            label="Number of Images"
-                        )
-                        
-                    with gr.Row():
-                        width = gr.Slider(
-                            minimum=256,
-                            maximum=1024,
-                            value=512,
-                            step=64,
-                            label="Width"
-                        )
-                        height = gr.Slider(
-                            minimum=256,
-                            maximum=1024,
-                            value=512,
-                            step=64,
-                            label="Height"
-                        )
-                    
-                    generate_btn = gr.Button("Generate Images", variant="primary")
                 
-                with gr.Column():
-                    output_gallery = gr.Gallery(
-                        label="Generated Images",
-                        show_label=True,
-                        elem_id="gallery"
-                    ).style(grid=2, height="auto")
-            
-            error_message = gr.Textbox(
-                label="Status",
-                visible=False
-            )
+                generate_btn = gr.Button("Generate Images", variant="primary")
             
-            # Add CSS for styling
-            gr.Markdown("""
-            <style>
-            .warning-banner {
-                background-color: #fff3cd;
-                border: 1px solid #ffeeba;
-                padding: 1rem;
-                margin-bottom: 1rem;
-                border-radius: 0.25rem;
-            }
-            </style>
-            """)
+            with gr.Column():
+                # Output gallery
+                output_gallery = gr.Gallery(
+                    label="Generated Images",
+                    show_label=True,
+                    elem_id="gallery"
+                ).style(grid=2, height="auto")
         
-        # Handle consent button clicks
-        def on_accept(user_id_val):
-            save_user_consent(user_id_val, True)
-            return {
-                warning_modal: gr.update(visible=False),
-                main_interface: gr.update(visible=True),
-                has_consented: True
-            }
-            
-        def on_decline(user_id_val):
-            save_user_consent(user_id_val, False)
-            return {
-                warning_modal: gr.update(visible=True),
-                main_interface: gr.update(visible=False),
-                has_consented: False
-            }
-        
-        accept_btn.click(
-            fn=on_accept,
-            inputs=[user_id],
-            outputs=[warning_modal, main_interface, has_consented]
+        # Error message display
+        error_message = gr.Textbox(
+            label="Status",
+            visible=False
         )
         
-        decline_btn.click(
-            fn=on_decline,
-            inputs=[user_id],
-            outputs=[warning_modal, main_interface, has_consented]
-        )
-        
-        # Handle image generation
-        def on_generate(prompt, style, num_images, width, height, has_consented_val):
-            if not has_consented_val:
-                return None, "Please accept the content warning first."
-            
+        # Handle generation
+        def on_generate(prompt, style, num_images, width, height):
             if not prompt.strip():
                 return None, "Please enter a prompt."
             
@@ -210,18 +120,18 @@ def create_interface():
         
         generate_btn.click(
             fn=on_generate,
-            inputs=[prompt, style, num_images, width, height, has_consented],
+            inputs=[prompt, style, num_images, width, height],
             outputs=[output_gallery, error_message]
         )
         
+        # Add usage instructions
         gr.Markdown("""
         ## How to Use
-        1. Accept the content warning to proceed
-        2. Enter a detailed description of the image you want to generate
-        3. Select a style from the dropdown menu
-        4. Adjust the number of images and dimensions as needed
-        5. Click 'Generate Images' and wait for the results
-        6. Generated images will be saved in the 'generated_images' folder
+        1. Enter a detailed description of the image you want to generate
+        2. Select a style from the dropdown menu
+        3. Adjust the number of images and dimensions as needed
+        4. Click 'Generate Images' and wait for the results
+        5. Generated images will be saved in the 'generated_images' folder
         
         Note: Generation time may vary depending on the complexity of your prompt.
         """)
@@ -229,6 +139,7 @@ def create_interface():
     return interface
 
 if __name__ == "__main__":
+    # Create and launch the interface
     app = create_interface()
     app.launch(
         share=True,