Update app.py

app.py

@@ -1,3 +1,238 @@
 import gradio as gr
+import os
+from datetime import datetime
+import json
+from http.cookies import SimpleCookie
 
-gr.load("models/LAYEK-143/FLUX_V0").launch()
+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):
+    """
+    Generate images based on the given prompt and parameters.
+    Returns list of generated image paths.
+    """
+    try:
+        model = gr.load("models/LAYEK-143/FLUX_V0")
+        images = []
+        for i in range(num_images):
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            filename = f"generated_{timestamp}_{i}.png"
+            
+            result = model.predict(
+                prompt=prompt,
+                style=style,
+                width=width,
+                height=height
+            )
+            
+            os.makedirs("generated_images", exist_ok=True)
+            save_path = os.path.join("generated_images", filename)
+            result.save(save_path)
+            images.append(save_path)
+            
+        return images
+    except Exception as e:
+        return str(e)
+
+def create_interface():
+    """Create and configure the Gradio interface"""
+    
+    STYLES = [
+        "Realistic",
+        "Artistic",
+        "Abstract",
+        "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")
+        
+        # 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
+                    )
+                    
+                    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"
+                        )
+                        
+                    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
+            )
+            
+            # 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>
+            """)
+        
+        # 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]
+        )
+        
+        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."
+            
+            if not prompt.strip():
+                return None, "Please enter a prompt."
+            
+            try:
+                images = generate_image(prompt, style, num_images, width, height)
+                if isinstance(images, str):  # Error occurred
+                    return None, images
+                return images, "Generation successful!"
+            except Exception as e:
+                return None, f"Error: {str(e)}"
+        
+        generate_btn.click(
+            fn=on_generate,
+            inputs=[prompt, style, num_images, width, height, has_consented],
+            outputs=[output_gallery, error_message]
+        )
+        
+        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
+        
+        Note: Generation time may vary depending on the complexity of your prompt.
+        """)
+    
+    return interface
+
+if __name__ == "__main__":
+    app = create_interface()
+    app.launch(
+        share=True,
+        enable_queue=True,
+        server_name="0.0.0.0",
+        server_port=7860
+    )