Create app.py

app.py

@@ -0,0 +1,76 @@
+import gradio as gr
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.chrome.options import Options
+import time
+
+# Function to generate image URL
+def generate_image(prompt):
+    # Configure Chrome options for headless mode
+    chrome_options = Options()
+    chrome_options.add_argument("--headless")
+    chrome_options.add_argument("--no-sandbox")
+    chrome_options.add_argument("--disable-dev-shm-usage")
+
+    # Initialize the WebDriver
+    driver = webdriver.Chrome(options=chrome_options)
+
+    try:
+        # Open the website
+        driver.get("https://gptimage.ai/")
+
+        # Wait for the textarea input field
+        input_field = WebDriverWait(driver, 10).until(
+            EC.presence_of_element_located(
+                (By.CSS_SELECTOR, "textarea.wp-content-visualizer-chat-input")
+            )
+        )
+        # Enter the prompt
+        input_field.send_keys(prompt)
+
+        # Wait for the "Generate Image" button and click it
+        generate_button = WebDriverWait(driver, 10).until(
+            EC.element_to_be_clickable(
+                (By.CSS_SELECTOR, "button.wp-content-visualizer-chat-send")
+            )
+        )
+        generate_button.click()
+
+        # Wait for the generated image
+        generated_image = WebDriverWait(driver, 20).until(
+            EC.presence_of_element_located(
+                (By.CSS_SELECTOR, "img.wp-content-visualizer-generated-image")
+            )
+        )
+
+        # Extract the image URL
+        image_url = generated_image.get_attribute("src")
+        return image_url
+
+    finally:
+        # Close the browser
+        driver.quit()
+
+# Gradio interface
+def gradio_interface(prompt):
+    try:
+        image_url = generate_image(prompt)
+        return f"Generated Image URL: {image_url}", image_url
+    except Exception as e:
+        return f"Error: {str(e)}", None
+
+# Create Gradio app
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs=gr.Textbox(label="Enter your prompt", placeholder="e.g., A serene mountain landscape at sunset"),
+    outputs=[gr.Textbox(label="Result"), gr.Image(label="Generated Image")],
+    title="Image Generator with Selenium",
+    description="Enter a prompt to generate an image using gptimage.ai."
+)
+
+# Launch the app
+if __name__ == "__main__":
+    iface.launch()