Update app.py

app.py

@@ -101,7 +101,7 @@ def encode_image_to_base64(image):
         return base64.b64encode(buffered.getvalue()).decode('utf-8')
 
 
-def generate_image_from_text(encoded_image,pos_prompt=None):
+def generate_image_from_text(encoded_image, seed, pos_prompt=None):
     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.'''
     encoded_str = encode_image_to_base64(encoded_image)
     if pos_prompt:
@@ -115,7 +115,7 @@ def generate_image_from_text(encoded_image,pos_prompt=None):
               },
               'imageGenerationConfig': {
                   "cfgScale": 8,
-                  "seed": 0,
+                  "seed": seed,
                   "width": 512,
                   "height": 512,
                   "numberOfImages": 1
@@ -131,7 +131,7 @@ def generate_image_from_text(encoded_image,pos_prompt=None):
           },
           'imageGenerationConfig': {
               "cfgScale": 8,
-              "seed": 0,
+              "seed": seed,
               "width": 512,
               "height": 512,
               "numberOfImages": 1
@@ -180,8 +180,8 @@ def generate(image, mc_resolution, formats=["obj", "glb"]):
     
     return mesh_path_obj.name, mesh_path_glb.name
 
-def run_example(image, do_remove_background, foreground_ratio, mc_resolution, text_prompt=None):
-    image_pil = generate_image_from_text(encoded_image=image, pos_prompt=text_prompt)
+def run_example(image, seed, do_remove_background, foreground_ratio, mc_resolution, text_prompt=None):
+    image_pil = generate_image_from_text(encoded_image=image, seed, pos_prompt=text_prompt)
     preprocessed = preprocess(image_pil, do_remove_background, foreground_ratio)
     mesh_name_obj, mesh_name_glb = generate(preprocessed, 256, ["obj", "glb"])
     return preprocessed, mesh_name_obj, mesh_name_glb
@@ -204,6 +204,7 @@ with gr.Blocks() as demo:
                     label="Text Prompt",
                     placeholder="Enter Positive Prompt"
                 )
+                seed = gr.Textbox(label="Random Seed", value=42)
                 processed_image = gr.Image(label="Processed Image", interactive=False, visible=False)
             with gr.Row():
                 with gr.Group():
@@ -242,7 +243,7 @@ with gr.Blocks() as demo:
                 
     submit.click(fn=check_input_image, inputs=[input_image]).success(
         fn=run_example,
-        inputs=[input_image, do_remove_background, foreground_ratio, mc_resolution, text_prompt],
+        inputs=[input_image, seed, do_remove_background, foreground_ratio, mc_resolution, text_prompt],
         outputs=[processed_image, output_model_obj, output_model_glb],
         # outputs=[output_model_obj, output_model_glb],
     )