Update app.py

app.py

@@ -1,17 +1,24 @@
 import gradio as gr
 from gradio_client import Client
 
+# Clients for image generation and swapping
 image_generation_client = Client("AP123/SDXL-Lightning")
 image_swapping_client = Client("aheedsajid/test")
 
+# Function to generate and swap images
 def generate_and_swap_image(prompt, source_image):
-    
     generated_image = image_generation_client.predict(prompt, "4-Step", api_name="/generate_image")
-    
     swapped_image = image_swapping_client.predict(source_image, generated_image, True, api_name="/predict")
-    
     return swapped_image
 
+# HTML code for the ad (replace with your actual ad code snippet)
+ad_code = """
+<div style="text-align: center;">
+    <iframe src="YOUR_AD_CODE_URL" width="300" height="250"></iframe>
+</div>
+"""
+
+# Gradio interface
 iface = gr.Interface(
     fn=generate_and_swap_image,
     inputs=[
@@ -20,7 +27,45 @@ iface = gr.Interface(
     ],
     outputs=gr.Image(label="Swapped Image"),
     title="AI Image with Any Face",
-    description="Please donate something so I can get a gpu server. As low as $1 also accepted<br>Generate free AI images with your face or any face.<br>Contact me to bypass NSFW for more creativity and freedom😉<br><b>Donate something so I can increase the GPU power/queue</b><br>[Click here to Donate](https://nowpayments.io/donation/aheed)<br>Contact me for bulk processing and more AI software at +92-332-4399819<br>Please support me in making better AI tools as I am a solo developer<br>[Click here to Donate](https://nowpayments.io/donation/aheed)<br><b>Do not duplicate this space without permission.</b>"
+    description="""
+        Please donate something so I can get a GPU server. As low as $1 also accepted<br>
+        Generate free AI images with your face or any face.<br>
+        Contact me to bypass NSFW for more creativity and freedom😉<br>
+        <b>Donate something so I can increase the GPU power/queue</b><br>
+        [Click here to Donate](https://nowpayments.io/donation/aheed)<br>
+        Contact me for bulk processing and more AI software at +92-332-4399819<br>
+        Please support me in making better AI tools as I am a solo developer<br>
+        [Click here to Donate](https://nowpayments.io/donation/aheed)<br>
+        <b>Do not duplicate this space without permission.</b>
+    """
 )
 
-iface.launch()
+# Adding the ad section
+iface = gr.Blocks()
+
+with iface:
+    gr.Markdown("# AI Image with Any Face")
+    gr.Markdown("""
+        Please donate something so I can get a GPU server. As low as $1 also accepted<br>
+        Generate free AI images with your face or any face.<br>
+        Contact me to bypass NSFW for more creativity and freedom😉<br>
+        <b>Donate something so I can increase the GPU power/queue</b><br>
+        [Click here to Donate](https://nowpayments.io/donation/aheed)<br>
+        Contact me for bulk processing and more AI software at +92-332-4399819<br>
+        Please support me in making better AI tools as I am a solo developer<br>
+        [Click here to Donate](https://nowpayments.io/donation/aheed)<br>
+        <b>Do not duplicate this space without permission.</b>
+    """)
+    
+    gr.HTML(ad_code)
+    
+    gr.Interface(
+        fn=generate_and_swap_image,
+        inputs=[
+            gr.Textbox(label="Enter your prompt (English)"),
+            gr.Image(label="Source Image", type="filepath"),
+        ],
+        outputs=gr.Image(label="Swapped Image")
+    )
+
+iface.launch()