Update app.py

app.py

@@ -7,7 +7,6 @@ import time
 
 repo = "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."
@@ -42,13 +41,12 @@ def infer(color_prompt, phone_type_prompt, design_prompt):
         else:
             raise Exception(f"API Error: {response.status_code}")
 
-# Function to save the design
-def save_design(image):
+def save_design(image_array):
+    image = Image.fromarray(image_array)
     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;
@@ -75,7 +73,6 @@ body::before {
 }
 """
 
-# JavaScript for text-to-speech and particles
 custom_js = """
 <script>
 document.addEventListener('DOMContentLoaded', function () {
@@ -105,7 +102,6 @@ document.addEventListener('DOMContentLoaded', function () {
 </script>
 """
 
-# Gradio interface
 with gr.Blocks(css=custom_css) as interface:
     gr.HTML(custom_js)
     gr.Markdown("# **AI Phone Cover Designer**")
@@ -155,5 +151,4 @@ with gr.Blocks(css=custom_css) as interface:
             Start designing today and bring your creative ideas to life!
             """)
 
-# Launch the app
-interface.launch(debug=True)
+interface.launch(debug=True)