Update app.py

app.py

@@ -7,32 +7,39 @@ from tqdm import tqdm
 import time
 
 repo = "artificialguybr/TshirtDesignRedmond-V2"
+
 def infer(color_prompt, dress_type_prompt, design_prompt, text):
+    # Improved prompt for higher accuracy
     prompt = (
-        f"A single {color_prompt} colored {dress_type_prompt} featuring a bold {design_prompt} design printed on the {dress_type_prompt}, hanging on a plain wall. The soft light and shadows, creating a striking contrast against the minimal background, evoking modern sophistication.")
-    full_prompt = f"{prompt}"
-
-    print("Generating image with prompt:", full_prompt)
+        f"A high-quality digital image of a {color_prompt} {dress_type_prompt}, "
+        f"featuring a {design_prompt} printed in sharp detail. The fabric has realistic texture, "
+        f"smooth folds, and accurate lighting. The design is perfectly aligned, with natural shadows "
+        f"and highlights, creating a photorealistic look."
+    )
+    
+    print("Generating image with prompt:", prompt)
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
-    #token = os.getenv("API_TOKEN")  # Uncomment and use your Hugging Face API token
-    headers = {
-        #"Authorization": f"Bearer {token}"
-    }
+    
+    headers = {}  # If API token needed, add here
+    
     payload = {
-        "inputs": full_prompt,
+        "inputs": prompt,
         "parameters": {
-            "negative_prompt": "(worst quality, low quality, normal quality, lowres, low details, oversaturated, undersaturated, overexposed, underexposed, grayscale, bw, bad photo, bad photography, bad art:1.4), (watermark, signature, text font, username, error, logo, words, letters, digits, autograph, trademark, name:1.2), (blur, blurry, grainy), morbid, ugly, asymmetrical, mutated malformed, mutilated, poorly lit, bad shadow, draft, cropped, out of frame, cut off, censored, jpeg artifacts, out of focus, glitch, duplicate, (airbrushed, cartoon, anime, semi-realistic, cgi, render, blender, digital art, manga, amateur:1.3), (3D ,3D Game, 3D Game Scene, 3D Character:1.1), (bad hands, bad anatomy, bad body, bad face, bad teeth, bad arms, bad legs, deformities:1.3), (Poor Fit, ill-fitting, awkward proportions, baggy where it shouldn't be, tight in wrong places, Bad Texture, low-quality fabric, wrinkled, rough texture, pixelated details, artificial shine, Cluttered Design, overly busy, too many patterns, excessive contrast, distracting elements, Outdated Fashion, old-fashioned, style(unless intentional), outdated trends, dull colors, Bad Composition, misaligned prints, asymmetrical in an unintentional way, weird placement of logos, Cheap Look, plastic-like fabric, low-quality print, faded colors, generic fast fashion, Unrealistic Details, floating textures, distorted logos, unnatural fabric folds, Unwanted Features, holes, rips, stains, unfinished seams, torn edges)",
+            # Optimized negative prompt
+            "negative_prompt": "low quality, artifacts, distorted, blurry, overexposed, underexposed, unrealistic texture, poor lighting, misaligned print, plastic-like fabric, grainy, washed-out colors, 3D render, cartoon, digital art, watermark, bad anatomy, malformed, cluttered design",
             "num_inference_steps": 30,
-            "scheduler": "DPMSolverMultistepScheduler"
+            "scheduler": "EulerAncestralDiscreteScheduler"  # Faster & more accurate scheduler
         },
     }
 
     error_count = 0
     pbar = tqdm(total=None, desc="Loading model")
+    
     while True:
         print("Sending request to API...")
         response = requests.post(api_url, headers=headers, json=payload)
         print("API response status code:", response.status_code)
+        
         if response.status_code == 200:
             print("Image generation successful!")
             return Image.open(BytesIO(response.content))
@@ -50,10 +57,10 @@ def infer(color_prompt, dress_type_prompt, design_prompt, text):
 iface = gr.Interface(
     fn=infer,
     inputs=[
-        gr.Textbox(lines=1, placeholder="Color Prompt"),         # color_prompt
-        gr.Textbox(lines=1, placeholder="Dress Type Prompt"),    # dress_type_prompt
-        gr.Textbox(lines=2, placeholder="Design Prompt"),        # design_prompt
-        gr.Textbox(lines=1, placeholder="Text (Optional)"),      # text
+        gr.Textbox(lines=1, placeholder="Color Prompt"),
+        gr.Textbox(lines=1, placeholder="Dress Type Prompt"),
+        gr.Textbox(lines=2, placeholder="Design Prompt"),
+        gr.Textbox(lines=1, placeholder="Text (Optional)"),
     ],
     outputs="image",
     title="Make your Brand",
@@ -62,4 +69,4 @@ iface = gr.Interface(
 )
 
 print("Launching Gradio interface...")
-iface.launch() 
+iface.launch()