Upload folder using huggingface_hub

main.py

@@ -6,6 +6,7 @@ from PIL import Image
 import requests
 import dotenv
 from io import BytesIO
+import random
 # from gradio_imageslider import ImageSlider
 
 dotenv.load_dotenv()
@@ -30,11 +31,54 @@ def get_gradio_component(class_type):
     }
     return component_map.get(class_type, gr.Textbox)  # Default to Textbox if not found
 
+with gr.Blocks() as demo:
+    gr.Markdown("""
+    # ComfyDeploy Gradio Interface
+
+    This is a Gradio interface for a ComfyDeploy workflow. You can interact with the deployed model using the inputs below.
+
+    To clone this workflow, visit: [ComfyDeploy Gradio Flux](https://www.comfydeploy.com/share/comfy-deploy-gradio-flux)
+
+    ## Example usage of ComfyDeploy SDK:
+
+    ```python
+    from comfydeploy import ComfyDeploy
+    import os
+
+    # Initialize the client
+    client = ComfyDeploy(bearer_auth=os.environ['API_KEY'])
+
+    # Run the model
+    inputs = {
+        'prompt': 'A beautiful landscape',
+        'negative_prompt': 'ugly, blurry',
+        'width': 512,
+        'height': 512
+    }
+    res = client.run.create(
+        request={
+            "deployment_id": deployment_id,
+            "inputs": inputs
+        }
+    )
+
+    # Get the results
+    run_id = res.object.run_id
+    result = client.run.get(run_id=run_id)
+    ```
+    """)
+
+def randomSeed(num_digits=15):
+    range_start = 10 ** (num_digits - 1)
+    range_end = (10**num_digits) - 1
+    return random.randint(range_start, range_end)
+
 # Function to update inputs
 def get_inputs():
     res = client.deployment.get_input_definition(id=deployment_id)
     input_definitions = res.response_bodies
     gradio_inputs = []
+    random_seeds = []
     for input_def in input_definitions:
         component_class = get_gradio_component(input_def.class_type)
         
@@ -57,21 +101,72 @@ def get_inputs():
         elif input_def.class_type == 'ComfyUIDeployExternalNumberInt':
             kwargs["precision"] = 0
             
+        if "seed" in input_def.input_id:
+            with gr.Row():
+                kwargs["value"] = randomSeed()
+                input = component_class(**kwargs, scale=6)
+                randomize_button = gr.Button("Randomize", size="sm")
+                def randomize_seed(input):
+                    return randomSeed()
+                randomize_button.click(fn=randomize_seed, inputs=input, outputs=input)
+                gradio_inputs.append(input)
+                random_seeds.append(input)
+            
         # print(kwargs)
+        else:
+            gradio_inputs.append(component_class(**kwargs))
         
-        gradio_inputs.append(component_class(**kwargs))
-        
-    return gradio_inputs, input_definitions
+    return gradio_inputs, input_definitions, random_seeds
 
 with gr.Blocks() as demo:
+    gr.Markdown("""
+    # ComfyDeploy Gradio Interface
+
+    This is a Gradio interface for a ComfyDeploy workflow. You can interact with the deployed model using the inputs below.
+
+    To clone this workflow, visit: [ComfyDeploy Gradio Flux](https://www.comfydeploy.com/share/comfy-deploy-gradio-flux)
+
+    ## Example usage of ComfyDeploy SDK:
+
+    ```python
+    from comfydeploy import ComfyDeploy
+    import os
+
+    # Initialize the client
+    client = ComfyDeploy(bearer_auth=os.environ['API_KEY'])
+
+    # Get input definitions
+    deployment_id = 'your_deployment_id'
+    res = client.deployment.get_input_definition(id=deployment_id)
+    input_definitions = res.response_bodies
+
+    # Run the model
+    inputs = {
+        'prompt': 'A beautiful landscape',
+        'negative_prompt': 'ugly, blurry',
+        'width': 512,
+        'height': 512
+    }
+    res = client.run.create(
+        request={
+            "deployment_id": deployment_id,
+            "inputs": inputs
+        }
+    )
+
+    # Get the results
+    run_id = res.object.run_id
+    result = client.run.get(run_id=run_id)
+    """)
+    
     with gr.Row():
         with gr.Column(scale=1):
             @gr.render()
             def update_inputs():
-                inputs, input_definitions = get_inputs()
+                inputs, input_definitions, random_seeds = get_inputs()
                 submit_button = gr.Button("Submit")
                 
-                async def main(*args):
+                async def main(*args, progress=gr.Progress()):
                     inputs = {input_def.input_id: arg for input_def, arg in zip(input_definitions, args)}
                     
                     for key, value in inputs.items():
@@ -96,6 +191,8 @@ with gr.Blocks() as demo:
                         if res.object is not None:
                             res2 = await client.run.get_async(run_id=res.object.run_id)
                             print("checking ", res2.object.progress, res2.object.live_status)
+                            progress_value = res2.object.progress if res2.object.progress is not None else 0
+                            progress(progress_value, desc=f"{res2.object.live_status}")
 
                             if res2.object is not None and res2.object.status == "success":
                                 # print(res2)
@@ -112,15 +209,18 @@ with gr.Blocks() as demo:
                                         outputs[1] += "\n\n" + "\n".join(output.data.text)
                                 break
                         await asyncio.sleep(2)
+                        
+                    for random_seed in random_seeds:
+                        random_seed.value = randomSeed()
 
-                    return outputs
+                    return outputs + random_seeds
                 
-                submit_button.click(fn=main, inputs=inputs, outputs=output_components)
+                submit_button.click(fn=main, inputs=inputs, outputs=output_components+random_seeds)
         
         with gr.Column(scale=1):
             output_components = [
                 gr.Gallery(),
-                gr.Textbox(label="Text Output")
+                gr.Textbox(label="Text Output"),
             ]
             
 if __name__ == "__main__":

requirements.txt

@@ -1,3 +1,4 @@
 comfydeploy==0.3.4
 gradio==4.41.0
-gradio_imageslider
+gradio_imageslider
+python-dotenv