Update app.py

app.py

@@ -97,7 +97,7 @@ def gen_pos_prompt(text):
   pos_prompt = json.loads(response.get('body').read())['results'][0]['outputText']
   return pos_prompt
 
-def generate_image_from_text(pos_prompt):
+def generate_image_from_text(pos_prompt, seed):
   new_prompt = gen_pos_prompt(pos_prompt)
   print(new_prompt)
   neg_prompt = '''Detailed, complex textures, intricate patterns, realistic lighting, high contrast, reflections, fuzzy surface, realistic proportions, photographic quality, vibrant colors, detailed background, shadows, disfigured, deformed, ugly, multiple, duplicate.'''
@@ -109,7 +109,7 @@ def generate_image_from_text(pos_prompt):
       'textToImageParams': {'text': new_prompt,
                             'negativeText': neg_prompt},
       'imageGenerationConfig': {"cfgScale":8,
-                                "seed":0,
+                                "seed":int(seed),
                                 "width":512,
                                 "height":512,
                                 "numberOfImages":1
@@ -159,9 +159,9 @@ def generate(image, mc_resolution, formats=["obj", "glb"]):
     
     return mesh_path_obj.name, mesh_path_glb.name
 
-def run_example(text_prompt, do_remove_background, foreground_ratio, mc_resolution):
+def run_example(text_prompt,seed do_remove_background, foreground_ratio, mc_resolution):
     # Step 1: Generate the image from text prompt
-    image_pil = generate_image_from_text(text_prompt)
+    image_pil = generate_image_from_text(text_prompt, seed)
     
     # Step 2: Preprocess the image
     preprocessed = preprocess(image_pil, do_remove_background, foreground_ratio)
@@ -188,6 +188,7 @@ with gr.Blocks() as demo:
                     elem_id="content_image",
                     visible=False  # Hidden since we generate the image from text
                 )
+                seed = gr.Number(value=0)
                 processed_image = gr.Image(label="Processed Image", interactive=False, visible=False)
             with gr.Row():
                 with gr.Group():
@@ -237,7 +238,7 @@ with gr.Blocks() as demo:
     #     )
     submit.click(fn=check_input_image, inputs=[text_prompt]).success(
         fn=run_example,
-        inputs=[text_prompt, do_remove_background, foreground_ratio, mc_resolution],
+        inputs=[text_prompt, seed, do_remove_background, foreground_ratio, mc_resolution],
         outputs=[processed_image, output_model_obj, output_model_glb],
         # outputs=[output_model_obj, output_model_glb],
     )