Updated: Removed Text Field

app.py

@@ -15,7 +15,7 @@ univin_model = os.environ.get("univin")
 client = genai.Client(api_key=api_key)
 
 CALL_INTERVAL = 2.0 / 10.0  # seconds
-dt_lock = Lock()
+_dt_lock = Lock()
 _last_call_time = 0.0
 
 def throttle():
@@ -23,7 +23,7 @@ def throttle():
     Ensures there's at least CALL_INTERVAL seconds between API calls.
     """
     global _last_call_time
-    with dt_lock:
+    with _dt_lock:
         now = time.time()
         elapsed = now - _last_call_time
         if elapsed < CALL_INTERVAL:
@@ -51,13 +51,14 @@ PROMPT_VARIATIONS = [
 ]
 
 # Detailed prompt variations for different backgrounds/angles
-def process_variation(variation, input_image, product_name):
+
+def process_variation(variation, input_image):
     # Throttle to respect API rate limit
     throttle()
 
     # Build text + image input
     text_input = (
-        f"Hi, this is a picture of a product. The name of the product is {product_name}.",
+        "Hi, this is a picture of a product.",
         variation
     )
 
@@ -79,19 +80,19 @@ def process_variation(variation, input_image, product_name):
     return None
 
 
-def generate_images(input_image, product_name):
+def generate_images(input_image):
     """
     Generate one image per prompt variation in parallel, respecting rate limits.
     """
     with concurrent.futures.ThreadPoolExecutor() as executor:
         futures = [
-            executor.submit(process_variation, var, input_image, product_name)
+            executor.submit(process_variation, var, input_image)
             for var in PROMPT_VARIATIONS
         ]
         return [f.result() for f in futures if f.result()]
 
 # Gradio UI setup with custom styling
-def build_interface():
+ def build_interface():
     custom_css = """
     #generate-button {
         background-color: #4A90E2;
@@ -128,11 +129,6 @@ def build_interface():
                 input_image = gr.Image(
                     type="pil", label="Upload Product Image", elem_id="upload-img"
                 )
-                product_name = gr.Textbox(
-                    label="Product Name",
-                    placeholder="Enter your product's name here",
-                    elem_id="product-name"
-                )
                 generate_button = gr.Button(
                     "Generate Images", variant="primary", elem_id="generate-button"
                 )
@@ -141,8 +137,7 @@ def build_interface():
                     """
                     **How to Use:**
                     1. Upload a clear photo of your product.
-                    2. Enter the product name.
-                    3. Click **Generate Images** and wait a few moments while images are processed within rate limits.
+                    2. Click **Generate Images** and wait a few moments while images are processed within rate limits.
                     """
                 )
         gallery = gr.Gallery(
@@ -155,7 +150,7 @@ def build_interface():
         )
         generate_button.click(
             fn=generate_images,
-            inputs=[input_image, product_name],
+            inputs=[input_image],
             outputs=gallery
         )
     return demo