Update app.py

app.py

@@ -20,23 +20,22 @@ MAX_IMAGE_SIZE = 1024
                )
 def infer(
     prompt,
-    negative_prompt,
+    
     seed,
     randomize_seed,
     width,
     height,
             # Enhancement
-    enhance=True,            # Auto-enhance prompt (default: False)
+    enhance=False,            # Auto-enhance prompt (default: False)
     
-    # Special Features
-    transparent=False,       # Transparent background (default: False)
+
     image=None,              # Input image URL for img2img
     
     # Privacy
     private=False,           # Private generation (default: False)
     nologo=False,            # Remove watermark (default: False)
     nofeed=False,            # Don't add to public feed (default: False)
-    safe=False  
+ 
     
 
 ):
@@ -58,7 +57,8 @@ def infer(
     url = f"https://image.pollinations.ai/prompt/{quote(prompt)}"
     # Customize the image size and model
     params = {"width": width, "height": height, "model": "flux","seed":seed}
-    
+    if image is not None:
+        params["image"]=image
     # Make the request
     response = requests.get(url, params=params, timeout=60)
     image_data = BytesIO(response.content)
@@ -70,7 +70,7 @@ def infer(
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
     "An astronaut riding a green horse",
-    "A delicious ceviche cheesecake slice",
+    "A delicious ceviche cheesecake slice"
 ]
 
 css = """
@@ -98,12 +98,6 @@ with gr.Blocks(css=css) as demo:
         result = gr.Image(label="Result", show_label=False)
 
         with gr.Accordion("Advanced Settings", open=False):
-            negative_prompt = gr.Text(
-                label="Negative prompt",
-                max_lines=1,
-                placeholder="Enter a negative prompt",
-                visible=False,
-            )
 
             seed = gr.Slider(
                 label="Seed",
@@ -132,22 +126,11 @@ with gr.Blocks(css=css) as demo:
                     value=1024,  # Replace with defaults that work for your model
                 )
 
-            # with gr.Row():
-            #     guidance_scale = gr.Slider(
-            #         label="Guidance scale",
-            #         minimum=0.0,
-            #         maximum=10.0,
-            #         step=0.1,
-            #         value=0.0,  # Replace with defaults that work for your model
-            #     )
-
-            #     num_inference_steps = gr.Slider(
-            #         label="Number of inference steps",
-            #         minimum=1,
-            #         maximum=50,
-            #         step=1,
-            #         value=2,  # Replace with defaults that work for your model
-                # )
+            with gr.Row():
+                enhance= gr.Checkbox(label="Randomize seed", value=False)
+                nologo=gr.Checkbox(label="nologo", value=False)
+                nofeed=gr.Checkbox(label="nofeed", value=False)
+
 
         gr.Examples(examples=examples, inputs=[prompt])
     gr.on(
@@ -155,11 +138,13 @@ with gr.Blocks(css=css) as demo:
         fn=infer,
         inputs=[
             prompt,
-            negative_prompt,
             seed,
             randomize_seed,
             width,
             height,
+            enhance,
+            nologo,            # Remove watermark (default: False)
+            nofeed=,            # Don't add to public feed (default: False)
                         
         ],
         outputs=[result, seed],