Update app.py

app.py

@@ -2,125 +2,54 @@ import gradio as gr
 from diffusers import DiffusionPipeline
 from PIL import Image
 import torch
-import os
 
-# Load Stable Diffusion XL pipeline and LoRA weights
+device = "cuda" if torch.cuda.is_available() else "cpu"
+dtype = torch.float16 if device == "cuda" else torch.float32
+variant = "fp16" if device == "cuda" else None
+
 pipe = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
-    torch_dtype=torch.float16,
-    variant="fp16"
-).to("cuda" if torch.cuda.is_available() else "cpu")
+    torch_dtype=dtype,
+    variant=variant
+).to(device)
+
 pipe.load_lora_weights("artificialguybr/TshirtDesignRedmond-V2")
 
-# Generate design based on prompts
 def infer(color_prompt, phone_type_prompt, design_prompt):
     prompt = (
         f"A single vertical {color_prompt} colored {phone_type_prompt} back cover featuring a bold {design_prompt} design on the front, hanging on the plain wall. The soft light and shadows, creating a striking contrast against the minimal background, evoking modern sophistication."
     )
-    print("Generating image with prompt:", prompt)
     image = pipe(prompt).images[0]
     message = "Design generated successfully!"
     return image, message
 
-# Function to save the design
 def save_design(image):
     file_path = "saved_design.png"
     image.save(file_path)
     return f"Design saved as {file_path}!"
 
-# Custom CSS for animations
-custom_css = """
-body {
-    font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
-    margin: 0;
-    padding: 0;
-    overflow: hidden;
-}
-body::before {
-    content: "";
-    position: fixed;
-    top: 0;
-    left: 0;
-    width: 100%;
-    height: 100%;
-    background: linear-gradient(-45deg, #ff9a9e, #fad0c4, #fbc2eb, #8fd3f4);
-    background-size: 400% 400%;
-    z-index: -1;
-    animation: gradientShift 15s ease infinite;
-}
-@keyframes gradientShift {
-    0% { background-position: 0% 50%; }
-    50% { background-position: 100% 50%; }
-    100% { background-position: 0% 50%; }
-}
-.carousel {
-    display: flex;
-    overflow-x: auto;
-    scroll-behavior: smooth;
-}
-.carousel .image-container {
-    display: inline-block;
-    text-align: center;
-    margin: 0 10px;
-}
-.carousel img {
-    max-height: 200px;
-    border-radius: 10px;
-    transition: transform 0.3s;
-}
-.carousel img:hover {
-    transform: scale(1.1);
-}
-.carousel .caption {
-    margin-top: 5px;
-    font-size: 14px;
-    color: #333;
-}
-"""
-
-with gr.Blocks(css=custom_css) as interface:
+with gr.Blocks() as interface:
     gr.Markdown("# **AI Phone Cover Designer**")
-    gr.Markdown("Create custom phone covers with AI. Save your designs for future use.")
-
-    with gr.Tabs():
-        with gr.Tab("Home"):
-            gr.Markdown("Welcome to the **AI Phone Cover Designer**! Use the 'Design' tab to start creating custom designs.")
-
-        with gr.Tab("Design"):
-            with gr.Row():
-                with gr.Column(scale=1):
-                    color_prompt = gr.Textbox(label="Color", placeholder="E.g., Red")
-                    phone_type_prompt = gr.Textbox(label="Mobile type", placeholder="E.g., iPhone, Samsung")
-                    design_prompt = gr.Textbox(label="Design Details", placeholder="E.g., Bold stripes with geometric patterns")
-                    generate_button = gr.Button("Generate Design")
-                    save_button = gr.Button("Save Design")
-                with gr.Column(scale=1):
-                    output_image = gr.Image(label="Generated Design")
-                    output_message = gr.Textbox(label="Status", interactive=False)
-
-            generate_button.click(
-                infer,
-                inputs=[color_prompt, phone_type_prompt, design_prompt],
-                outputs=[output_image, output_message],
-            )
-            save_button.click(
-                save_design,
-                inputs=[output_image],
-                outputs=output_message,
-            )
-
-        with gr.Tab("About"):
-            gr.Markdown("""
-            ## About AI Phone Cover Maker
-            The **AI Phone Cover Maker** is a cutting-edge tool designed to help users create personalized phone cover designs quickly and easily. 
-            Powered by AI, it uses advanced image generation techniques to craft unique, high-quality designs for any mobile device.
-            
-            ### Features:
-            - Create custom designs using simple prompts.
-            - Generate designs for various phone models.
-            - Save your designs for future use.
-            
-            Start designing today and bring your creative ideas to life!
-            """)
+    with gr.Row():
+        with gr.Column(scale=1):
+            color_prompt = gr.Textbox(label="Color")
+            phone_type_prompt = gr.Textbox(label="Mobile Type")
+            design_prompt = gr.Textbox(label="Design Details")
+            generate_button = gr.Button("Generate Design")
+            save_button = gr.Button("Save Design")
+        with gr.Column(scale=1):
+            output_image = gr.Image(label="Generated Design")
+            output_message = gr.Textbox(label="Status", interactive=False)
+
+    generate_button.click(
+        infer,
+        inputs=[color_prompt, phone_type_prompt, design_prompt],
+        outputs=[output_image, output_message],
+    )
+    save_button.click(
+        save_design,
+        inputs=[output_image],
+        outputs=output_message,
+    )
 
-interface.launch(debug=True)
+interface.launch(debug=True)