Upload 23 files

.gitattributes

@@ -1,35 +1,35 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

Prompts.txt

@@ -0,0 +1,138 @@
+Lift Up Vag Flash.
+She lifts the entire bottom portion of her clothing upward to just above her vagina area. All of the bottom fabric is now fully scrunched and bunched tightly in her hands and not visible anywhere else, completely exposing her shaved, beautiful vagina. Any other objects that were in her hands must be removed. Tiny gap of space between her legs. Everything else in the scene remains exactly the same as before.
+
+Same Place Sitting Tease.
+She sits on the floor in the same room with the same clothes on but the top of them are pulled down to be under her exposed breasts which must be firm and perky and standing up straight with nipples hard and pointing right at the camera without any  sagging. and the bottom of her clothes are pulled up in a way to not cover her vagina. but with her knees facing straight towards me but her vagina is facing at me, same exact identical face with all of it's exact features and same exact original expression with no changes except her eyes are looking directly at the camera, she uses her hands to grip both sides of her vulva and fully part/spread both sides so that her vagina insides are visible and as far as they be parted  as those same hands also push against her inner thighs visibly, everything else the same
+
+Same Place Shh Tease.
+Her left hand's index finger is hovering away from her vagina and she is pointing it directly at her vagina hole while she also holds her right hand's index finger on her mouth saying shh but still smiling extremely sexually suggestively and invitingly, everything else stays the same!
+
+Same Place Shh Tease 2.
+Her left hand is being used to touch her vaginal clitoris without covering the whole vagina while she also holds her right hand's index finger on her mouth saying shh but still smiling extremely sexually suggestively and invitingly, vagina fully visible between her now slightly more spread legs. tonge is pressed between her teeth in a sexually suggestive way. Exact same identical face with all the exact same identical features. everything else stays the same!
+
+1 Leg Up Inviting Tease.
+She holds 1 leg far up in the air, to fully show vagina and, vagina is facing at me, everything else the same
+
+Same Place Squatting Frontal.
+Same exact identical face with all the same exact identical features. 
+head is turned almost fully around naturally to look directly into my eyes. She is being fully penetrated by a small but wide penis coming from the bottom of the picture attached to a man who is no visible who has his legs stretched under her. her mouth wide open with her lips making an o shape. She looks like she is imdeep in love and dreaming in a fantasy. She is very close to the camera sitting on the penis that is now fully inside her vagina hole, showing how it parts her vulva from its penetration.
+remove the objects in the way. She touches her clitoris with just one of her hand's fingers and her other hand stays is grabbing the penis in her vagina.
+everything else is the exact same!
+
+Dick Holding Pleasure Tease.
+She is squatting up in the air close to the camera with her legs only partially spread. With her right hand, she is eagerly, firmly, and tightly gripping a small penis that enters the frame from the bottom of the image; the penis belongs to an otherwise completely unseen man whose legs are stretched out flat beneath her. 
+She hovers directly above the tip of the penis. the head of the penis is touching / pressing against her vaginal area. 
+She uses her left hand to reach underneath her clothing as she has her fingers deeply inside her vagina, they disappear inside her.
+Her head is turned almost completely around in a natural way so she can stare directly into the viewer's eyes.  Both of her breasts are fully exposed outside of her clothing, with visibly hardened nipples. her open mouth—lips shaped into a perfect "O"—Remove all flowers from the image. Everything else must remain exactly the same.
+
+Dick Holding Almost BJ.
+She leans forward with her mouth wide open and lips in an O shape so that her mouth is closer to the penis but not touching it. her tongue is visible only behind the penis pressed against it. She is still firmly grasping the penis the same exact way. Close up of her face. Same exact identical face with all the same features. Everything else is the same
+
+Same Place Partial BJ.
+Same environment.
+She is now on her stomach as far back enough as needed and bending forward enough so that the penis that is coming up from the bottom of the photo that is attached to an unseen man is now all the way in her mouth. put the penis much deeper in her mouth.  You can see her butt visibly behind her head as well which has smooth skin and arches for the cheeks. same exact identical face with all the same exact identical features. Everything else is the exact same.
+
+Side View Oral.
+She is wearing the same clothing the same exact way. Her left hand is sexually touching her vagina between her fully spread legs. Her vagina is fully visible and you can see she put fingers inside of it's hole. 
+The same sized but smaller penis from the original photo is now  all the way deep inside of her mouth and you can see that it is in her throat because there is a bulge from it on her throat. The very short penis is coming from the right edge of the photo. Her lips are tightly closed surrounding the bottom part of the penis. She is squatting at a side angle in a way that you can see what I said she is doing with her vagina. The camera angle makes that possible. Her right hand holds the penis and pulls it more into her mouth. Same exact identical face with exact same facial features. her eyes are open and rolled back in pleasure. Remove the man from the photo and keep only the shaft of the penis coming from the right edge of the photo.
+
+Remove Her Clothes.
+no clothing, naked, perfectly accurate, everything else is the exact same
+
+Full Body Nudity.
+no clothing, naked, full body, perfectly accurate, same exact identical face with all the same exact idential features, everything else is the exact same
+
+Secret Silent Sex.
+close up, same exact face, same clothes but breasts fully out with hard nipples and vagina fully showing as the clothes just barely don't cover it. she is leaning very far forwards very close to the camera, full breasts visible and perfect, sitting on a very small penis fully penetrating all the way to the farthest depths of inside her vagina, coming up from an unseen man with his legs stretched underneath her, everything else the same, vulva fully  parted from the penis, she also firmly presses her right hand's index finger up to the center of her lips as if to say shh  but with her mouth wide open, tongue partially showing inside of it with her lips making a small partially closed O shape.
+everything else stays the same!
+
+Rear View Anal.
+She is turned around  facing away from the camera, squatting and bent over a chair she holds that is far away from her but in front of her. He clothing is still the same and each part is the same but her breasts are visible outside and the bottom of her clothing is bunched up above her waist. She uses the mans penis that is in her ass to support her. Her feet are touching the ground. her butt is higher in the air so that her ass hole is penetrated by the penis. Same exact identical face as the original woman in the original photo. Her head is turned almost completely around naturally and her eyes are seen looking directly into the camera with her mouth wide open and her lips making an O shape. She is pushing her body backwards into man's penis from the bottom of the photo to get the penis further than it already is inside her ass hole. the ass is higher than the vagina and the vagina cannot be seen. her butt is angled upwards so her vagina is close up facing right at me. Make the ass hole itself higher than normal where her tailbone is instead
+
+Spread On Bed Pose.
+no clothes, laying on her girly bed in a just slightly dark room, same exact identical face with all the same exact identical features. her legs are much thinner skinnier and shorter and spread apart to fully show her tiny tight beautiful vagina, identical face and expression, vagina pointed at me. everything else the same!
+
+Come Get It Pose.
+no clothes, laying on her girly bed in a slightly dimly lit room, her legs are much thinner skinnier and shorter and spread apart to fully show her tiny tight beautiful vagina, exact same identical face and all the exact same facial features, vagina pointed at me, her left hand's index finger is hovering away from her vagina and she is pointing it directly at her vagina hole while she also holds her right hand's index finger on her mouth saying shh behind a partial kissy face. but still looking right into my eyes  extremely sexually suggestively and excitingly and invitingly, slightly smiling looking excited to finally have me all to herself. close up of vagina. full breasts are visible with hardened nipples. everything else stays the same!
+
+Secret Silent Room Sex.
+close up, same exact identical face with the exact same identical features, on a couch, no clothes, she is leaning very far forwards very close to the camera, full breasts visible and perfect, sitting on a very small penis fully penetrating all the way to the farthest depths of inside her vagina, coming up from an unseen man with his legs stretched underneath her, everything else the same, vulva fully  parted from the penis, she also holds her right hand's index finger on her mouth saying shh but with a sexy secretive sneaky look   extremely sexually but distracted by the pleasure, mouth slightly open with lips making an o shape from the immense pleasure, dark bedroom background, everything else stays the same!
+
+Front View Couch Sex.
+front view, close up of her face, she is on a very small  penis that cannot be fully seen because it is already fully all the way inside her vagina, coming up from a man that cannot be seen on the bottom of the photo underneath her, same exact face, fully naked, on a couch, full perky breasts visible and perfect,  same exact identical face and person in general but her mouth is open with lips in a closed small surprised o shape with her eyes looking at me but slightly to the top right, dark bedroom background, top part of her sexy cocktail dress is pulled down to be underneath her visible breasts and the bottom part of her dress is pulled upwards still around her legs but not covering her vagina, Exact same identical face with all the exact same identical features. everything else the same.
+
+Rear View Couch Sex.
+she is squatting all the way down at the bottom center of the photo with her vagina fully penetrated by a short little penis also coming from all the way at the bottom center of the photo. 
+back view, fully naked on a couch with a slight dark bedroom background. the man's legs are seen completely closed together but out straight between her legs. the entire penis is hidden inside of her vagina. she wears a sexy cocktail dress that doesn't cover her vagina. her breasts are also outside of the dress with hardened nipple showing. Her legs are much thinner and petite. both sides of her fat  vulva are noticeably spread apart from the penis and her head is turned almost completely around naturally as far as she can to look directly into my eyes, her mouth is open with lips in a closed small surprised o shape. her butt is pointing up at the camera so you can see the intimate details of her full penetrated vagina. Exact same identical face with all the exact same identical features. she is also reaching behind her back and using only  her little hand to grab and spreading more open her right butt cheek so the penis is seen even deeper inside of her. only the penis of the man is visible. everything else is the exact same!
+
+Balcony Risky Sex.
+On a balcony at night, leaning over the railing, small penis inside her vagina from behind coming from an unseen man beyond the edge of the photo, dress bunched at waist, wind blowing, one leg lifted, vagina fully visible and dripping. Same exact identical face with all the same exact identical features turned back with risky excited smile and open mouth. Everything else the exact same!
+
+Bathroom Sink Sex.
+Sitting on the bathroom sink facing the camera, legs spread wide on the edges, small penis fully penetrating her vagina coming from an unseen man beyond the edge of the photo, water running in background, both hands pulling her breasts up and together while vagina is  slightly naturally stretched visibly. Same exact identical face with all the same exact identical features moaning with pure pleasure looking straight at viewer. Everything else stays the exact same!
+
+Bed Arching Back Orgasm Pose.
+No clothes, lying on her girly bed with back arched high, legs spread maximally and pulled back, small penis fully embedded in her vagina coming from an unseen man beyond the edge of the photo from below with the man's legs under her, her hands pinching her hard nipples. Same exact identical face with all the same exact identical features showing maximum ecstasy with eyes rolled back and tongue hanging out. Vagina is fully visible and glistening. Everything else stays the same!
+
+Bed Doggy Style Front Turned.
+On all fours on the bed facing away but head fully turned back to camera naturally, small penis penetrating her vagina deeply from behind coming from an unseen man beyond the edge of the photo, one hand reaching under to spread her vulva lips around the shaft showing insertion. Same exact identical face with all the same exact identical features with eyes locked on viewer in dreamy loving pleasure. Her butt cheeks are spread naturally. Everything else the exact same!
+
+Bent Over Table Anal-Vaginal Tease.
+Bent over a table in the same room, clothing bunched at waist, small penis deep in her ass coming from an unseen man beyond the edge of the photo but legs spread so her vagina is prominently visible and dripping below the penetration, she reaches back with both hands to spread her ass cheeks and vulva simultaneously. Same exact identical face with all the same exact identical features turned around to the camera with wide O mouth in overwhelming pleasure. Everything else the exact same!
+
+Car Backseat Ride.
+In the backseat of a car, straddling facing camera, small penis buried inside her vagina coming from an unseen man beyond the edge of the photo, windows slightly fogged, dress pulled down and up, hands on the headrest behind her pushing her breasts forward. Same exact identical face with all the same exact identical features in maximum arousal looking at viewer. Everything else the exact same!
+
+Chair Ride Facing Camera.
+Sitting on a chair facing the camera, small penis fully inside her vagina coming from an unseen man beyond the edge of the photo, legs draped over the chair arms spreading her extremely wide, both hands pulling her vulva lips apart showing the penetration and her insides. Breasts fully exposed with hard nipples. Same exact identical face with all the same exact identical features staring directly at the viewer with raw sexual hunger. Everything else stays the exact same!
+
+Close Up Face With Vaginal Spread And Penis Tease.
+Extreme close up on her face and upper body but vagina visible in lower frame, she holds the small penis coming from an unseen man beyond the edge of the photo with one hand pressing it against her clitoris while the other hand spreads her vulva, mouth open tongue out licking her lips suggestively. Same exact identical face with all the same exact identical features but with maximum seductive arousal. Everything else stays the same!
+
+Couch Lap Dance Penetration.
+On the couch, she straddles facing the camera in lap position with small penis fully inside her vagina coming from an unseen man beyond the edge of the photo, dress pulled up and down exposing breasts and vagina, hands on her own breasts squeezing while bouncing slightly implied. Same exact identical face with all the same exact identical features, mouth open in surprised pleasure O shape looking slightly up at camera. Everything else the exact same!
+
+Intense Missionary Leg Lift.
+No clothes, lying on her back on the girly bed with both legs pulled up high and spread extremely wide, small penis buried to the hilt inside her vagina coming from an unseen man beyond the edge of the photo from below with the unseen man's legs stretched underneath. Her hands grip her own thighs pulling them apart even more so the slightly naturally stretched vulva and deep penetration are perfectly visible. Same exact identical face with all the same exact identical features showing pure overwhelming ecstasy, eyes half-closed in bliss, mouth wide open in a perfect moaning O shape. Everything else stays the exact same!
+
+Kitchen Counter Penetration.
+Bent over the kitchen counter, legs spread wide, small penis deep inside her vagina from behind coming from an unseen man beyond the edge of the photo, dress pulled up, breasts hanging and pressed against the counter, one hand reaching back spreading herself. Same exact identical face with all the same exact identical features turned back with an expression of being fucked senseless in ecstasy. Everything else the exact same!
+
+Kneeling Frontal Invitation With Fingers.
+Fully naked, kneeling on the floor close to camera with legs spread, using both index fingers to hook and slightly naturally pull her vagina lips apart extremely wide showing the deepest insides while a small penis tip coming from an unseen man beyond the edge of the photo presses against the entrance from below. Same exact identical face with all the same exact identical features smiling extremely sexually suggestively with pure invitation in her eyes. Everything else stays the same!
+
+Lotus Position Deep Intimate.
+Fully naked in lotus position on the bed facing the camera, small penis completely inside her vagina coming from an unseen man beyond the edge of the photo, legs wrapped around the unseen man but spread enough for full view, her hands on his shoulders (unseen) while she grinds. Same exact identical face with all the same exact identical features showing intense loving pleasure, mouth open, eyes dreamy. Everything else the exact same!
+
+Mirror Reflection Tease.
+Standing in front of a large mirror, bent slightly forward, small penis entering her vagina from behind coming from an unseen man beyond the edge of the photo while she watches herself in the mirror, one hand spreading her vulva, the other pinching a nipple. Same exact identical face with all the same exact identical features reflected in the mirror staring back at the viewer with maximum seductive arousal. Everything else stays the exact same!
+
+Overhead Squat Ride Ecstasy.
+From an overhead camera angle, she squats deeply onto the small penis fully inside her vagina coming from an unseen man beyond the edge of the photo, vulva slightly naturally stretched wide and parted, her hands reaching down to rub her clitoris around the shaft. Same exact identical face with all the same exact identical features turned upwards to the camera with an expression of being lost in intense sexual fantasy and love. Everything else the exact same!
+
+Prone Bone Close Up.
+Lying flat on her stomach on the bed, ass slightly raised, small penis deep inside her vagina from behind coming from an unseen man beyond the edge of the photo, one hand reaching back to slightly naturally spread her ass and vulva so the slightly naturally stretched hole and penetration are clearly visible. Same exact identical face with all the same exact identical features turned to the side looking back at the camera with blissful orgasm face. Everything else the exact same!
+
+Reverse Cowgirl Deep Squat.
+Fully naked, squatting with her back to the camera but head turned almost 180 degrees naturally to stare straight into the viewer's eyes, small penis completely swallowed by her vagina coming from an unseen man beyond the edge of the photo, vulva lips gripping the base tightly. One hand reaches back spreading her ass cheek while the other rubs her clitoris around the shaft. Same exact identical face with all the same exact identical features lost in maximum sexual pleasure, tongue slightly out. Everything else the exact same!
+
+Shower Wet Sex.
+Standing in the shower under running water, one leg wrapped around the unseen man, small penis deep inside her vagina coming from an unseen man beyond the edge of the photo, water cascading over her body, one hand spreading her vulva around the shaft. Same exact identical face with all the same exact identical features showing wet, slippery, maximum sexual ecstasy. Everything else stays the exact same!
+
+Sideways Spooning Vaginal Sex.
+She lies on her side in the same room, one leg lifted high allowing full view of her vagina being penetrated by the small penis coming from the side edge of the photo attached to unseen man, her top hand slightly naturally spreading her vulva further while bottom hand grips the penis base. Same exact identical face with all the same exact identical features, mouth in perfect O shape staring invitingly at camera. Everything else the exact same!
+
+Standing Bent Over Window Sex.
+She bends forward pressing her hands against the window, ass pushed out, small penis thrusting deep into her vagina from behind coming from an unseen man beyond the edge of the photo, legs spread wide so her dripping vagina is fully visible between her thighs. Clothing bunched at waist exposing breasts and vagina. Same exact identical face with all the same exact identical features turned back to the camera with desperate arousal and open moaning mouth. Everything else stays the exact same!
+
+Standing Carry Position Tease.
+She is held up by unseen man (his arms and legs partially visible), legs wrapped around but spread to show vagina penetrated by small penis standing coming from an unseen man beyond the edge of the photo, her hands around his neck but since unseen focus on her, vagina fully visible slightly naturally stretched. Same exact identical face with all the same exact identical features smiling invitingly with tongue between teeth. Everything else the exact same!
+
+Wall Leaning Deep Penetration Tease.
+She leans back against the wall with her legs spread wide and one knee raised high, the small penis  coming from an unseen man beyond the edge of the photo below is thrusting deep into her vagina causing visible stretching slightly naturally and her juices flowing freely. Same exact identical face with all the same exact identical features looking straight at the camera with eyes full of desperate arousal and lips parted in a silent scream of pleasure. Everything else the exact same!
+
+Wall Pressed Standing Sex.
+Pressed face-first against the wall, ass pushed out, small penis buried deep in her vagina coming from an unseen man beyond the edge of the photo, one leg lifted high to the side for maximum visibility of the penetration and her dripping vulva. Same exact identical face with all the same exact identical features pressed against the wall but turned enough to lock eyes with the viewer in pure lust. Everything else stays the exact same!
+
+Yoga Mat Deep Stretch Sex.
+On a yoga mat in downward dog position but modified, small penis penetrating her vagina deeply from behind coming from an unseen man beyond the edge of the photo, one leg lifted high to the side for extreme spread and visibility of penetration. Same exact identical face with all the same exact identical features looking back between her legs at the camera in overwhelming pleasure. Everything else stays the exact same!

README.md

@@ -1,15 +1,14 @@
----
-title: Qwen Image Edit NSFW
-emoji: 🎨
-colorFrom: purple
-colorTo: pink
-sdk: gradio
-sdk_version: 5.0.0
-app_file: app.py
-pinned: false
-license: apache-2.0
----
-
-# Qwen Image Edit NSFW - Zero GPU
-
-Identical to the original picgen application, optimized for Hugging Face Spaces Zero GPU.
+1. Connect to vps
+2. Clone this repo
+3. cd to /picgen
+4. Run bash setup.sh
+
+To run later:
+
+ssh-keygen -f '/root/.ssh/known_hosts' -R 'ip_address'
+
+ssh root@ip_address
+
+cd picgen
+
+python3 app.py --share

app.py

@@ -2,7 +2,9 @@ import gradio as gr
 import numpy as np
 import random
 import torch
-import spaces
+# spaces import REMOVED — ZeroGPU is HuggingFace-only infrastructure, not needed locally
+
+
 import gc
 
 from safetensors.torch import load_file
@@ -40,23 +42,26 @@ NSFW_WEIGHTS_LOCAL_PATH = os.path.join(MODELS_DIR, "rapid-aio", "v23", "Qwen-Rap
 # Detect operating system
 IS_WINDOWS = platform.system() == "Windows"
 IS_LINUX = platform.system() == "Linux"
-print("Operating System: {platform.system()}")
+print(f"Operating System: {platform.system()}")
+
+
+def encode_image(pil_image):
+    import io
+    buffered = io.BytesIO()
+    pil_image.save(buffered, format="PNG")
+    return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
 # --- Clear GPU memory from previous runs ---
 def clear_vram():
     gc.collect()
+    torch.cuda.empty_cache()
     if torch.cuda.is_available():
-        torch.cuda.empty_cache()
         torch.cuda.synchronize()
 
-# Clear GPU at startup if available (Zero GPU will skip this safely)
-try:
-    print("Clearing GPU memory from previous runs...")
-    clear_vram()
-    if torch.cuda.is_available():
-        torch.cuda.reset_peak_memory_stats()
-except:
-    print("GPU not available at startup (normal for Zero GPU)")
+print("Clearing GPU memory from previous runs...")
+clear_vram()
+if torch.cuda.is_available():
+    torch.cuda.reset_peak_memory_stats()
 
 # --- Model Loading ---
 dtype = torch.bfloat16
@@ -333,7 +338,13 @@ def add_starter_image(starter_num, current_images):
     return list(current_images) + [img]
 
 # --- Main Inference Function (with hardcoded negative prompt) ---
-@spaces.GPU
+# ---------------------------------------------------------------------------
+# CHANGE 4 of 4: @spaces.GPU() decorator REMOVED
+# That decorator allocates a GPU from HuggingFace's shared ZeroGPU pool.
+# On a local VPS the GPU is always available — the decorator is not needed
+# and would crash without the `spaces` library installed.
+# The entire function body below is 100% identical to the original.
+# ---------------------------------------------------------------------------
 def infer(
     images,
     prompt,
@@ -377,13 +388,12 @@ def infer(
     if images is not None:
         for item in images:
             try:
-                # item is now a filepath string
-                if isinstance(item, str):
-                    pil_images.append(Image.open(item).convert("RGB"))
-                elif isinstance(item, dict) and 'name' in item:
-                    pil_images.append(Image.open(item['name']).convert("RGB"))
-                elif isinstance(item, Image.Image):
-                    pil_images.append(item.convert("RGB"))
+                if isinstance(item[0], Image.Image):
+                    pil_images.append(item[0].convert("RGB"))
+                elif isinstance(item[0], str):
+                    pil_images.append(Image.open(item[0]).convert("RGB"))
+                elif hasattr(item, "name"):
+                    pil_images.append(Image.open(item.name).convert("RGB"))
             except Exception:
                 continue
 
@@ -472,8 +482,10 @@ with gr.Blocks(css=css) as demo:
                     "Front View Couch Sex",
                     "Rear View Couch Sex"
                 ],
+                value=None,
                 interactive=True,
-                show_label=False
+                show_label=False,
+                allow_custom_value=False
             )
             run_button_top = gr.Button("Edit!", variant="primary", size="sm")
         
@@ -516,8 +528,10 @@ with gr.Blocks(css=css) as demo:
                     "Full Penetration Attempt 9",
                     "Full Penetration Attempt 10"
                 ],
+                value=None,
                 interactive=True,
-                show_label=False
+                show_label=False,
+                allow_custom_value=False
             )
             run_button_top2 = gr.Button("Edit!", variant="primary", size="sm")
         
@@ -582,8 +596,10 @@ with gr.Blocks(css=css) as demo:
                     "Full Penetration Attempt 9 (Couple)",
                     "Full Penetration Attempt 10 (Couple)"
                 ],
+                value=None,
                 interactive=True,
-                show_label=False
+                show_label=False,
+                allow_custom_value=False
             )
             run_button_top3 = gr.Button("Edit!", variant="primary", size="sm")
         
@@ -626,8 +642,10 @@ with gr.Blocks(css=css) as demo:
                     "Side Fuck",
                     "Kissing Sex"
                 ],
+                value=None,
                 interactive=True,
-                show_label=False
+                show_label=False,
+                allow_custom_value=False
             )
             run_button_top4 = gr.Button("Edit!", variant="primary", size="sm")
         
@@ -641,10 +659,11 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 input_images = gr.Gallery(label="Input Images", 
                                           show_label=False, 
-                                          type="filepath")
+                                          type="pil", 
+                                          interactive=True)
 
             with gr.Column():
-                result = gr.Gallery(label="Result", show_label=False, type="filepath")
+                result = gr.Gallery(label="Result", show_label=False, type="pil", interactive=False)
                 use_output_btn = gr.Button("↗️ Use as input", variant="secondary", size="sm", visible=False)
 
         with gr.Row():
@@ -707,7 +726,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=2048,
                     step=8,
-                    value=256,
+                    value=None,
                 )
                 
                 width = gr.Slider(
@@ -715,8 +734,20 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=2048,
                     step=8,
-                    value=256,
+                    value=None,
                 )
+            
+            fullscreen_toggle = gr.Checkbox(
+                label="Full Screen Mode",
+                value=False,
+                info="Expand image boxes to full page width"
+            )
+            
+            keyboard_toggle = gr.Checkbox(
+                label="Disable On-Screen Keyboard",
+                value=False,
+                info="Prevent keyboard from appearing on mobile devices"
+            )
 
         # gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=False)
 
@@ -962,5 +993,84 @@ with gr.Blocks(css=css) as demo:
     start3_btn.click(fn=lambda imgs: add_starter_image(3, imgs), inputs=[input_images], outputs=[input_images])
     start4_btn.click(fn=lambda imgs: add_starter_image(4, imgs), inputs=[input_images], outputs=[input_images])
 
+    # Preset handler for negative prompts
+    def update_negative_prompt(preset):
+        presets = {
+            "None": "",
+            "Full Penetration (Vaginal)": "partial insertion, visible shaft, shallow, tip only, halfway, partially inserted, pulling out, not fully inside",
+            "Full Penetration (Oral)": "shallow, tip only, not deepthroat, partially in mouth, lips at tip, halfway, visible shaft outside mouth",
+            "Full Penetration (Anal)": "partial insertion, visible shaft, shallow, tip only, halfway, partially inserted, not fully inside, pulling out",
+        }
+        return presets.get(preset, "")
+    
+    # REMOVED - negative_preset dropdown no longer exists
+    
+    # Fullscreen toggle handler
+    fullscreen_toggle.change(
+        fn=None,
+        inputs=[fullscreen_toggle],
+        outputs=None,
+        js="""
+        (fullscreen) => {
+            const style = document.getElementById('fullscreen-style') || document.createElement('style');
+            style.id = 'fullscreen-style';
+            if (fullscreen) {
+                style.textContent = `
+                    .gradio-container { max-width: 100vw !important; padding: 0 !important; }
+                    .contain { max-width: 100% !important; }
+                    .wrap { max-width: 100% !important; }
+                    #component-0 { max-width: 100% !important; }
+                `;
+            } else {
+                style.textContent = '';
+            }
+            if (!document.getElementById('fullscreen-style')) {
+                document.head.appendChild(style);
+            }
+            return fullscreen;
+        }
+        """
+    )
+    
+    # Keyboard toggle handler
+    keyboard_toggle.change(
+        fn=None,
+        inputs=[keyboard_toggle],
+        outputs=None,
+        js="""
+        (disable) => {
+            setTimeout(() => {
+                const style = document.getElementById('keyboard-style') || document.createElement('style');
+                style.id = 'keyboard-style';
+                if (disable) {
+                    style.textContent = `
+                        input[type="text"]:not([role="combobox"]), 
+                        input[type="number"], 
+                        textarea:not([role="combobox"]) { 
+                            -webkit-user-select: none !important;
+                            user-select: none !important;
+                            pointer-events: none !important;
+                        }
+                    `;
+                    document.querySelectorAll('input[type="text"]:not([role="combobox"]), input[type="number"], textarea:not([role="combobox"])').forEach(el => {
+                        el.setAttribute('readonly', 'readonly');
+                        el.setAttribute('inputmode', 'none');
+                    });
+                } else {
+                    style.textContent = '';
+                    document.querySelectorAll('input[type="text"], input[type="number"], textarea').forEach(el => {
+                        el.removeAttribute('readonly');
+                        el.removeAttribute('inputmode');
+                    });
+                }
+                if (!document.getElementById('keyboard-style')) {
+                    document.head.appendChild(style);
+                }
+            }, 100);
+            return disable;
+        }
+        """
+    )
+
 if __name__ == "__main__":
-    demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False, show_api=False)
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)

commands.txt

@@ -0,0 +1,30 @@
+# Image Editor Service Commands
+
+## Setup (run once)
+sudo cp picgen.service /etc/systemd/system/
+sudo systemctl daemon-reload
+sudo systemctl enable picgen
+
+## Start the service
+sudo systemctl start picgen
+
+## Stop the service
+sudo systemctl stop picgen
+
+## Restart the service
+sudo systemctl restart picgen
+
+## Check status
+sudo systemctl status picgen
+
+## View logs (live)
+sudo journalctl -u picgen -f
+
+## View logs (last 100 lines)
+sudo journalctl -u picgen -n 100
+
+## Disable service (prevent auto-start on boot)
+sudo systemctl disable picgen
+
+## Kill the process manually (if needed)
+sudo pkill -f "python3.*app.py"

fix.bat

@@ -0,0 +1,13 @@
+@echo off
+echo === Fixing Picgen Installation ===
+
+echo Clearing Python cache...
+for /d /r . %%d in (__pycache__) do @if exist "%%d" rd /s /q "%%d"
+del /s /q *.pyc 2>nul
+
+echo Reinstalling dependencies...
+pip3 install -r requirements.txt --force-reinstall
+
+echo === Fix Complete ===
+echo Run: start.bat
+pause

fix.sh

@@ -0,0 +1,20 @@
+#!/bin/bash
+set -e
+
+echo "=== Fixing Picgen Installation ==="
+
+cd /root/picgen
+
+echo "Clearing Python cache..."
+find . -type d -name "__pycache__" -exec rm -rf {} + 2>/dev/null || true
+find . -type f -name "*.pyc" -delete 2>/dev/null || true
+
+echo "Reinstalling dependencies..."
+pip3 install -r requirements.txt --break-system-packages --ignore-installed --force-reinstall
+
+echo "Copying service file..."
+cp picgen.service /etc/systemd/system/
+systemctl daemon-reload
+
+echo "=== Fix Complete ==="
+echo "Run: python3 app.py"

ftp.py

@@ -0,0 +1,43 @@
+import subprocess
+import sys
+import os
+
+# 1. Automatic Dependency Check & Install
+try:
+    from pyftpdlib.authorizers import DummyAuthorizer
+    from pyftpdlib.handlers import FTPHandler
+    from pyftpdlib.servers import FTPServer
+except ImportError:
+    print("pyftpdlib not found. Installing...")
+    subprocess.check_call(["apt", "install", "-y", "python3-pip"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "pyftpdlib", "--break-system-packages"])
+    from pyftpdlib.authorizers import DummyAuthorizer
+    from pyftpdlib.handlers import FTPHandler
+    from pyftpdlib.servers import FTPServer
+
+def main():
+    # 2. Setup credentials
+    authorizer = DummyAuthorizer()
+    
+    # Replace 'YOUR_PASSWORD' with your desired password
+    # "." ensures it shares the current folder and all subfolders
+    # perm='elradfmw' provides full read, write, and list permissions
+    authorizer.add_user("admin", "YOUR_PASSWORD", ".", perm="elradfmw")
+
+    # 3. Setup handler
+    handler = FTPHandler
+    handler.authorizer = authorizer
+    handler.banner = "FTP Server is ready."
+
+    # 4. Start server
+    # 0.0.0.0 makes it public; 2121 avoids needing root privileges for port 21
+    address = ("0.0.0.0", 2121)
+    server = FTPServer(address, handler)
+
+    print(f"FTP server active at: {os.getcwd()}")
+    print("Accessible on port 2121")
+    server.serve_forever()
+
+if __name__ == "__main__":
+    main()
+    

manual install.txt

@@ -0,0 +1,7 @@
+
+Requirements.txt as background to avoid broken ssh pipe                      
+nohup pip3 install -r requirements.txt --break-system-packages --ignore-installed --timeout=300 --retries=10 > pip-install.log 2>&1 &
+
+
+Then monitor it:
+tail -f pip-install.log

new.txt

@@ -0,0 +1,114 @@
+Merge Together 1
+make them both standing next to eachother facing the camera. She is to the right of him. Her face and body are perfectly exactly the same. she wears a cute sundress that is loose at the bottom  and he wears a stylish open button down shirt with a white t shift under it. his penis is exactly the same. 
+
+Merge Together 2
+They are both standing next to each other facing the camera. She is on his right. She wears the same exact clothing (or none if she isn't wearing any) that is loose at the bottom and he wears a stylish open button down shirt with a white t-shirt under it. Both maintain their exact same identical original faces with all features preserved. his penis is every way the same, unchanged.
+
+Naked Together
+Add the woman from the first photo to be standing on the right of the man in the second photo, ensuring the woman is fully naked wearing no clothes at all. She has the exact same face, facial expression, etc as her original photo. Keep his penis exactly the way it is without altering it. Both maintain their exact same identical original faces with all features preserved. Her face and facial expression and angle of it must be exactly identical.es preserved.
+
+Near Dick 1
+fully zoomed out so she is fully in view. the woman from the first picture looking exactly 100% the same with same exact face and facial features and expression are exactly the same. She is completely moved from the 1st photo to be in the 2nd photo so that she is sitting behind that man's separate penis. she has no pants on and no underwear on and nothing covering her vagina. she has no clothing around her waste or under it. Her legs are spread and she is sitting in a way so that her entire vagina is fully visible. The  man's penis is exactly the same. the man is still wearing the same clothing. Remove all other things from the first picture and only use the background/environment of the second picture. 
+
+Near Dick 2
+She has no clothing around her waste or under it and she is squatting near the penis from photo 2. we can see her full vagina. you can still see her face. she is looking at the camera the exact same way as her original photo. The penis from photo 2 is in her hand. she has the exact very young looking same face and everything else about her face and head and hair is the exact same. everything else the same from photo 2. 
+
+Near Dick 3
+zoomed out to show her full body. She has no clothing on and she is now only in photo 2 near the man's penis. you can still see her face. she is looking at the camera the exact same way as her original photo. she holds the man's penis from photo 2 is in her hand as it is attached to the man in the same exact way. she has the exact very young looking same face and everything else about her face and head and hair is the exact same. everything else the same from photo 2. 
+
+Near Dick 4
+zoomed out to show her full body. She has no clothing on from her waist down and she is now perfectly embedded only in photo 2 near the man's penis. His penis looks exactly the same from the original photo. her legs are open showing her vagina. you can still see her face. she is looking at the camera the exact same way as her original photo. she holds the man's penis from photo 2 is in her hand as it is attached to the man in the same exact way. she has the exact very young looking same face and everything else about her face and head and hair is the exact same. everything else the same from photo 2. 
+
+Near Dick 5
+Sitting legs spread behind the penis pressed against her vagina
+She is sitting on the man in the second photo that has everything the exact same. she has her legs spread behind that exact same penis and she is pushing her vagina against the back of it which is wedging the penis in between her now visibly spread vulva. everything else from the second photo is exactly the same. the environment of the first photo is removed and replaced with the exact environment of the second photo. the penis is exactly the same identical penis in the same exact position etc.  
+
+Shh Holding Dick
+She is now holding his penis with only one of her hands as she looks sexually horny at the camera and her other hand has it's index finder on her lips as she says shh. she is demanding sex. everything else is the exact same.
+
+Both Near Dick
+both women from the first picture looking exactly 100% the same are now only in the second photo, near the legs and penis of the man in the second photo.  the women's faces are fully preserved/unchanged as well as their clothing etc. The man's penis is exactly the same. the man is still wearing the same clothing. Remove all other things from the first photo and only use the background/environment of the second photo. they are in the second photo only. do not include  anything from the first photo except the 2 women.
+
+Both Holding Dick 1
+both women are how holding his penis in their hands and looking at the camera. 
+Both women maintain their exact same identical original faces with all the exact same identical features. Everything else stays the exact same!
+
+Both Holding Dick 2
+both women from the first picture looking exactly 100% the same are now only in the second photo, near the legs and penis of the man in the second photo.  the women's faces are fully preserved/unchanged as well as their clothing etc. The man's penis is exactly the same. the man is still wearing the same clothing. Remove all other things from the first photo and only use the background/environment of the second photo. they are in the second photo only. do not include  anything from the first photo except the 2 women.
+
+Single BJ
+the woman is now bent forward down fully sucking his entire shaft of his penis now hidden inside of her mouth with her lips sealed around the circumference of the lowest part of the shaft. She does all of that while sexually touching her own vagina.
+She maintains her exact same identical original face with all the exact same identical features. Everything else stays the exact same!
+
+Single Anal
+the woman is now turned around and on his penis that is now fully penetrated deep inside of her ass. The entire shaft of his penis now hidden inside of her ass which has the entire penis all the way to the lowest part of the shaft. She is turned so you can see her exact same original face. She does this while sexually touching her own vagina.
+The woman maintains her exact same identical original face with all the exact same identical features. Everything else stays the exact same!
+
+Left 1 BJ
+the woman on the left   is bent forward down fully sucking his entire shaft of his penis now hidden inside of her mouth with her lips sealed around the circumference of the lowest part of the shaft. . the other woman  is watching while sexually touching her own vagina.
+Both women maintain their exact same identical original faces with all the exact same identical features. Everything else stays the exact same!
+
+Left 1 Anal
+the woman on the left is now turned around and on his penis that is now fully penetrated deep inside of her ass.  the entire shaft of his penis now hidden inside of her ass which has the entire penis all the way to the lowest part of the shaft. she is turned so you can see her exact same original face. the other woman  is watching while sexually touching her own vagina.
+Both women maintain their exact same identical original faces with all the exact same identical features. Everything else stays the exact same!
+
+Left 1 Vag On Knee
+the woman on the left is now sitting on his right knee with her vagina showing. the other woman is on his left knee with her vagina showing.
+Both women maintain their exact same identical original faces with all the exact same identical features. Everything else stays the exact same!
+
+Fucked after 1
+the woman on the left   is now turned around and on his penis that is now fully penetrated deep inside of her ass.  the entire shaft of his penis now hidden inside of her ass which has the entire penis all the way to the lowest part of the shaft. she is turned so you can see her exact same original face. the other woman  is watching while sexually touching her own vagina.
+Both women maintain their exact same identical original faces with all the exact same identical features. Everything else stays the exact same!
+
+Fucked after 2
+the woman on the right   is now turned around and on his penis that is now fully penetrated deep inside of her ass.  the entire shaft of his penis now hidden inside of her ass which has the entire penis all the way to the lowest part of the shaft. she is turned so you can see her exact same original face. the other woman  is watching while sexually touching her own vagina.
+Both women maintain their exact same identical original faces with all the exact same identical features. Everything else stays the exact same!
+
+Penetration 1
+the penis is now only fully penetrated inside her vagina. she is still looking at the camera but with her mouth open and her lips in the shape of an o. she no longer holds a penis because it is now sticking into her vagina. the penis is the same in every way.
+
+Penetration 2
+the same exact original  penis is now only fully penetrated inside her vagina. she is still looking at the camera but with her mouth open and her lips in the shape of an o. she no longer holds a penis because the same one is now sticking into her vagina. the penis is the same in every way. the mans legs fully outstretched and together under her. the penis is exactly the same. 
+
+Penetration 3
+the penis is now only fully penetrated all the way inside her vagina as she has 1 leg in the air and looks dreamily into his eyes. the full shaft of the penis now inserted and hidden inside of her. with her mouth open and her lips in the shape of an o. her face is exactly the same
+
+Shower Sex Together
+They are both standing in the shower, she has one leg lifted near the man in the photo, his penis deep inside her vagina, one hand spreading her vulva around the shaft.  zoomed out for full view 
+
+Kissing Handjob
+the man is now sitting down on the ground  facing the bottom right corner of the photo  and the woman is squatting higher in the air beside him, almost hip to hip.
+the man now has his penis out of his pants and
+the woman's is fully gripping it tightly around the full circumference of it's shaft in one of her hands while her other hand is behind his head while she passionately kisses him on the lips.
+
+
+BJ
+The woman from the second photo eagerly has her hand that is the closest to him holding his penis with her fingers wrapped fully around the shaft. The position of the rest of her body remains unchanged. The man from the first photo remains completely unchanged. She remains in the exact same position next to him. You can fully see her hand's fingers wrapped completely around it. the penis stays exactly the way it is in the original photo. Keep his penis exactly the way it is without altering it. Both the man and woman preserve their exact same faces and facial features.
+
+BJ 2
+the woman from the first photo squats down next to the man from the second photo so that her face is at the man's penis which is now all the way deep inside her mouth with the full shaft of it hidden inside of her mouth which is closed around it.  Her right hand holds his penis and pulls it more into her mouth. Her lips are tightly closed surrounding the bottom of his penis.  Her left hand also sexually touches her vagina between her fully spread legs with fingers inside. the man's head is hidden. is moved first so that all of these things are possible.  her legs are spread and you can fully see her vagina in the photo and also see her fingering her vagina. Keep her exact same face and facial features preserved, her eyes open and rolled back in pleasure.
+
+Bed BJ 1
+the man from the second photo and the woman from the first photo are on a bed in a girly room. the man is has his eyes closed. her exact face is perfectly preserved. she has his full penis inside of her mouth. she has a sexually aroused smiling and sneaky look on her face because she is doing this without his permission while he is sleeping. The full shaft of the penis is buried deep inside of her mouth so it is hidden from the view of the camera. zoomed out view. his penis is exactly where it should be on his body.
+
+Bed BJ 2
+the man from the second photo and the woman from the first photo are on a bed in a girly room. her exact face is perfectly preserved. she has his full penis inside of her mouth as she has a freaky sexual deviant look on her face. she looks at the penis as it is all the way inside so that the full shaft of it is buried deep inside and hidden from view of the camera.
+
+Bed Sex 1
+the man from the second photo and the woman from the first photo are on a bed in a girly room. they are having passionate sex. You can see his penis all the way inside of her vagina and her face with wide open mouth and lips making an o shape. both their faces perfectly preserved. they both look down at the penis going into her vagina all the way inside so that the full shaft of it is buried deep inside and hidden from view of the camera.
+
+Bed Sex2
+the man from the second photo and the woman from the first photo are on a bed in a girly room.  You can see the man's penis is all the way inside of the woman's vagina. Her face is perfectly preserved. Her eyes look down at the penis going into her vagina. The man's penis is all the way inside of her vagina so that the full shaft of it is buried deep inside and hidden from view of the camera.
+
+Bed Sex 3
+the male in the second photo has his penis fully penetrating the female from the first photo in her vagina, on a bed. the female's face is fully preserved as the exact same with all the same exact features.
+
+Bed Sex 4
+the woman from the first photo is in the exact same pose/position. the man in the second photo is laying on her same bed with his legs outwads between hers and his penis penetrating her vagina. everything about them is the same
+
+Side Fuck
+She lies on her side, one leg lifted high allowing full view of her vagina being penetrated by his entire penis. Her mouth is in perfect O shape.  Both maintain their exact same identical original faces. they both are looking down at the penetration. Everything else stays the same!
+
+Kissing Sex
+she drops her butt and vagina more onto his penis so that his penis is fully inserted deeply into her vagina as she rides his penis (they are having sex) as they open-mouth French kiss passionately
+

optimization.py

@@ -0,0 +1,85 @@
+"""
+"""
+from typing import Any
+from typing import Callable
+from typing import ParamSpec
+from torchao.quantization import quantize_
+from torchao.quantization import Float8DynamicActivationFloat8WeightConfig
+# spaces import REMOVED — spaces.GPU / spaces.aoti_* are HuggingFace ZeroGPU-only APIs.
+# They do not exist outside of HF infrastructure and would crash on a local VPS.
+# Replaced below with standard torch.compile() which gives equivalent or better
+# performance on a dedicated GPU (RTX 4090 / PRO 6000 Blackwell).
+import torch
+from torch.utils._pytree import tree_map
+
+
+P = ParamSpec('P')
+
+
+TRANSFORMER_IMAGE_SEQ_LENGTH_DIM = torch.export.Dim('image_seq_length')
+TRANSFORMER_TEXT_SEQ_LENGTH_DIM = torch.export.Dim('text_seq_length')
+
+TRANSFORMER_DYNAMIC_SHAPES = {
+    'hidden_states': {
+        1: TRANSFORMER_IMAGE_SEQ_LENGTH_DIM,
+    },
+    'encoder_hidden_states': {
+        1: TRANSFORMER_TEXT_SEQ_LENGTH_DIM,
+    },
+    'encoder_hidden_states_mask': {
+        1: TRANSFORMER_TEXT_SEQ_LENGTH_DIM,
+    },
+    'image_rotary_emb': ({
+        0: TRANSFORMER_IMAGE_SEQ_LENGTH_DIM,
+    }, {
+        0: TRANSFORMER_TEXT_SEQ_LENGTH_DIM,
+    }),
+}
+
+
+INDUCTOR_CONFIGS = {
+    'conv_1x1_as_mm': True,
+    'epilogue_fusion': False,
+    'coordinate_descent_tuning': True,
+    'coordinate_descent_check_all_directions': True,
+    'max_autotune': True,
+    'triton.cudagraphs': True,
+}
+
+
+def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
+    # -------------------------------------------------------------------------
+    # CHANGE 1 of 1: spaces.GPU / spaces.aoti_capture / spaces.aoti_compile /
+    # spaces.aoti_apply REMOVED.
+    #
+    # What the original did on HuggingFace ZeroGPU:
+    #   1. spaces.aoti_capture()  — traced one forward pass to record all inputs
+    #   2. torch.export.export()  — exported the transformer to a portable graph
+    #   3. spaces.aoti_compile()  — compiled via torch inductor with INDUCTOR_CONFIGS
+    #   4. spaces.aoti_apply()    — swapped the live module for the compiled artifact
+    #
+    # Local replacement:
+    #   torch.compile() with the same inductor backend and the same INDUCTOR_CONFIGS
+    #   achieves the identical goal — persistent kernel fusion, cudagraphs, and
+    #   coordinate-descent autotuning — without any HF-specific infrastructure.
+    #   The compiled transformer is set back onto the pipeline in-place, so every
+    #   call to pipeline() automatically uses the optimised version, exactly as
+    #   the original spaces.aoti_apply() did.
+    #
+    # NOTE: The first inference call after optimize_pipeline_() will be slow
+    # while torch.compile() traces and compiles the kernels. Subsequent calls
+    # are fast. This is the same warm-up behaviour as the original AOTI path.
+    #
+    # To enable Float8 quantisation (commented out in the original too), uncomment:
+    #   quantize_(pipeline.transformer, Float8DynamicActivationFloat8WeightConfig())
+    # -------------------------------------------------------------------------
+
+    # quantize_(pipeline.transformer, Float8DynamicActivationFloat8WeightConfig())
+
+    pipeline.transformer = torch.compile(
+        pipeline.transformer,
+        backend="inductor",
+        options=INDUCTOR_CONFIGS,
+        fullgraph=False,   # False = safe for models with dynamic control flow
+        dynamic=True,      # honours the dynamic sequence-length dims declared above
+    )

picgen.service

@@ -0,0 +1,20 @@
+[Unit]
+Description=Picgen Image Editor Application
+After=network.target
+
+[Service]
+Type=simple
+User=root
+WorkingDirectory=/root/picgen
+Environment="PYTHONUNBUFFERED=1"
+Environment="HF_HOME=/root/.cache/huggingface"
+Environment="PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True"
+Environment="CUDA_LAUNCH_BLOCKING=0"
+ExecStart=/usr/bin/python3 /root/picgen/app.py
+Restart=always
+RestartSec=10
+StandardOutput=append:/root/picgen/picgen.log
+StandardError=append:/root/picgen/picgen.log
+
+[Install]
+WantedBy=multi-user.target

requirements.txt

@@ -18,7 +18,8 @@ Pillow
 numpy
 
 # UI and utilities
-huggingface_hub<1.0.0
+gradio
+huggingface_hub
 tqdm
 typing-extensions
 

setup.bat

@@ -0,0 +1,32 @@
+@echo off
+echo === Picgen Image Editor Windows Setup ===
+
+echo Creating cache directory...
+if not exist "%USERPROFILE%\.cache\huggingface" mkdir "%USERPROFILE%\.cache\huggingface"
+
+echo Installing PyTorch with CUDA 12.4 support...
+pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu124
+
+echo Installing Python dependencies...
+pip3 install -r requirements.txt
+
+echo Installing Hugging Face CLI...
+pip3 install huggingface_hub[cli]
+
+echo Creating local model directories...
+if not exist "models\Qwen-Image-Edit-2511" mkdir "models\Qwen-Image-Edit-2511"
+if not exist "models\rapid-aio\v23" mkdir "models\rapid-aio\v23"
+
+echo === Model Download ===
+echo Models will be downloaded automatically on first run.
+echo To pre-download models now, run:
+echo   huggingface-cli download Qwen/Qwen-Image-Edit-2511 --local-dir models/Qwen-Image-Edit-2511
+echo   huggingface-cli download Phr00t/Qwen-Image-Edit-Rapid-AIO --include "v23/Qwen-Rapid-AIO-NSFW-v23.safetensors" --local-dir models/rapid-aio
+
+echo Verifying GPU accessibility...
+python -c "import torch; print(f'CUDA Available: {torch.cuda.is_available()}'); print(f'GPU Device: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else \"None\"}'); print(f'Total VRAM: {torch.cuda.get_device_properties(0).total_memory / (1024**3):.1f}GB' if torch.cuda.is_available() else 'N/A')"
+
+echo.
+echo === Setup Complete ===
+echo To start the app, run: start.bat
+pause

setup.sh

@@ -0,0 +1,60 @@
+#!/bin/bash
+set -e
+
+echo "=== Picgen Image Editor VPS Setup ==="
+
+echo "Creating cache directory..."
+mkdir -p /root/.cache/huggingface
+
+echo "Installing system dependencies..."
+apt-get update && apt-get install -y python3-pip python3-venv ffmpeg wget git git-lfs bc
+
+echo "Installing PyTorch with CUDA 12.4 support..."
+pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu124 --break-system-packages --ignore-installed
+
+echo "Installing Python dependencies..."
+pip3 install -r requirements.txt --break-system-packages --ignore-installed
+
+echo "Installing Hugging Face CLI..."
+pip3 install huggingface_hub[cli] --break-system-packages
+
+echo "Creating local model directories..."
+mkdir -p models/Qwen-Image-Edit-2511
+mkdir -p models/rapid-aio/v23
+chmod -R 777 models
+
+echo "=== Model Download ==="
+echo "Models will be downloaded automatically on first run."
+echo "To pre-download models now, run:"
+echo "  huggingface-cli download Qwen/Qwen-Image-Edit-2511 --local-dir models/Qwen-Image-Edit-2511"
+echo "  huggingface-cli download Phr00t/Qwen-Image-Edit-Rapid-AIO --include 'v23/Qwen-Rapid-AIO-NSFW-v23.safetensors' --local-dir models/rapid-aio"
+
+echo "Verifying GPU accessibility..."
+python3 -c "
+import torch
+print(f'CUDA Available: {torch.cuda.is_available()}')
+if torch.cuda.is_available():
+    print(f'GPU Device: {torch.cuda.get_device_name(0)}')
+    total_vram = torch.cuda.get_device_properties(0).total_memory / (1024**3)
+    print(f'Total VRAM: {total_vram:.1f}GB')
+else:
+    print('GPU Device: None (CPU mode)')
+"
+
+echo "=== Setup Complete ==="
+echo ""
+echo "Setting up systemd service..."
+cp picgen.service /etc/systemd/system/
+systemctl daemon-reload
+systemctl enable picgen
+systemctl start picgen
+
+echo ""
+echo "Service commands:"
+echo "  systemctl start picgen   - Start picgen"
+echo "  systemctl stop picgen    - Stop picgen"
+echo "  systemctl status picgen  - Check status"
+echo "  systemctl restart picgen - Restart picgen"
+echo ""
+echo "To run manually: python3 app.py"
+echo "The app will be accessible at: http://0.0.0.0:7860"

ssh.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+ssh-keygen -f '/root/.ssh/known_hosts' -R '194.93.48.120'
+ssh -o StrictHostKeyChecking=no -o ServerAliveInterval=60 -o ServerAliveCountMax=3 root@194.93.48.120
+echo ""
+echo "To connect manually, run:"
+echo "ssh root@194.93.48.120"

start.bat

@@ -0,0 +1,4 @@
+@echo off
+echo === Starting Picgen ===
+python app.py
+pause

start.sh

@@ -0,0 +1,11 @@
+#!/bin/bash
+
+sudo cp /root/picgen/picgen.service /etc/systemd/system/
+sudo systemctl daemon-reload
+sudo systemctl enable picgen.service
+sudo systemctl start picgen.service
+
+echo "✓ picgen service started!"
+echo ""
+echo "Check status: sudo systemctl status picgen"
+echo "View logs: sudo journalctl -u picgen -f"

stop.bat

@@ -0,0 +1,5 @@
+@echo off
+echo === Stopping Picgen ===
+taskkill /F /IM python.exe /FI "WINDOWTITLE eq *app.py*"
+echo Picgen stopped.
+pause

stop.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+
+sudo systemctl stop picgen.service
+sudo systemctl disable picgen.service
+
+echo "✓ picgen service stopped!"

switch_to_picgen.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+sudo systemctl stop vidgen.service
+sudo systemctl disable vidgen.service
+sudo cp /root/picgen/picgen.service /etc/systemd/system/
+sudo systemctl daemon-reload
+sudo systemctl enable picgen.service
+sudo systemctl start picgen.service
+echo "Switched from vidgen to picgen!"
+echo "Check status: sudo systemctl status picgen"

switch_to_vidgen.sh

@@ -0,0 +1,28 @@
+#!/bin/bash
+set -e
+
+echo "=== Switching from picgen to vidgen ==="
+
+# Stop and disable picgen
+sudo systemctl stop picgen.service 2>/dev/null || true
+sudo systemctl disable picgen.service 2>/dev/null || true
+
+# Kill any remaining Python processes
+sudo pkill -9 python3 2>/dev/null || true
+sudo pkill -9 python 2>/dev/null || true
+sleep 2
+
+# Copy vidgen service file
+sudo cp /root/vidgen/vidgen.service /etc/systemd/system/
+
+# Reload systemd
+sudo systemctl daemon-reload
+
+# Enable and start vidgen
+sudo systemctl enable vidgen.service
+sudo systemctl start vidgen.service
+
+echo "✓ Switched to vidgen!"
+echo ""
+echo "Check status: sudo systemctl status vidgen"
+echo "View logs: sudo journalctl -u vidgen -f"