Update app.py

app.py

@@ -1,3 +1,4 @@
+from run import process
 import time
 import subprocess
 import os
@@ -30,6 +31,7 @@ def inference(img):
     bgra = cv2.cvtColor(img, cv2.COLOR_RGBA2BGRA)
     inputpath = f"input_{index}.jpg"
     cv2.imwrite(inputpath, bgra)
+
     outputpath = f"out_{index}.jpg"
     index += 1
     print(time.strftime("START!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
@@ -37,49 +39,35 @@ def inference(img):
     print(time.strftime("Finish!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
     return output
 
-def update_status(img):
-    return inference(img), gr.update(value="Processing complete!")
+title = "Undress AI"
+description = "⛔ Input photos of people, similar to the test picture at the bottom, and undress pictures will be produced. You may have to wait 30 seconds for a picture. 🔞 Do not upload personal photos 🔞 There is a queue system. According to the logic of first come, first served, only one picture will be made at a time. Must be able to at least see the outline of a human body ⛔"
+
+examples = [
+    ['input.png', 'Test'],
+    ['input.jpg', 'Test'],
+]
 
-def init_interface(request: gr.Request):
-    query_params = request.query_params
-    bg_color = query_params.get('bg_color', 'rgb(17, 24, 39)')
-    image_height = query_params.get('image_height', '90%')
-    css = f"""
-    body {{
-        background-color: {bg_color};
-        color: white;
-        overflow: hidden;
-    }}
-    .gradio-container {{
-        background-color: {bg_color} !important;
-        border: none !important;
-    }}
-    .image-container {{
-        height: {image_height} !important;
-        display: flex;
-        align-items: center;
-        justify-content: center;
-    }}
-    .image-container img {{
-        width: auto !important;
-        height: 100% !important;
-    }}
-    footer {{
-        display: none !important;
-    }}
-    """
+css = """
+body {
+    background-color: rgb(17, 24, 39);
+    color: white;
+    overflow: hidden; /* Prevent scrolling */
+}
+.gradio-container {
+    background-color: rgb(17, 24, 39) !important;
+    border: none !important;
+}
+footer {display: none !important;} /* Hide footer */
+"""
 
-    with gr.Blocks(css=css) as demo:
-        with gr.Column():
-            with gr.Row(elem_id="image-container"):
-                image_input = gr.Image(type="numpy", label="Upload Image")
-            process_button = gr.Button("Process Image")
+with gr.Blocks(css=css) as demo:
+    with gr.Column():
+        image_input = gr.Image(type="numpy", label="Upload Image")
+        process_button = gr.Button("Process Image")
 
-        process_button.click(update_status, inputs=image_input, outputs=[image_input])
-    
-    return demo
+    def update_status(img):
+        return inference(img), gr.update(value="Processing complete!")
 
-with gr.Blocks() as outer_demo:
-    outer_demo.load(init_interface)
+    process_button.click(update_status, inputs=image_input, outputs=[image_input])
 
-outer_demo.launch()
+demo.launch()