Upload folder using huggingface_hub

gradio_sketch2image.py

@@ -380,5 +380,38 @@ with gr.Blocks(css="style.css") as demo:
     run_button.click(fn=run, inputs=inputs, outputs=outputs, api_name=False)
     image.change(run, inputs=inputs, outputs=outputs, queue=False, api_name=False)
 
+# Create a simple drawing interface
+def drawing_interface():
+    with gr.Blocks(css="style.css") as draw_interface:
+        gr.Markdown("# Drawing Interface")
+        draw_canvas = gr.Image(source="canvas", tool="color-sketch", type="pil", label="Draw your sketch here", image_mode="L", invert_colors=True, brush_radius=4)
+        generate_button = gr.Button("Generate Image")
+        output_image = gr.Image(label="Generated Image")
+        
+        generate_button.click(
+            fn=run,
+            inputs=[draw_canvas, gr.Textbox(), gr.Textbox(value=styles[DEFAULT_STYLE_NAME]), gr.Dropdown(choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME), gr.Slider(value=DEFAULT_SEED), gr.Slider(value=0.4)],
+            outputs=[output_image]
+        )
+        
+    return draw_interface
+
+# Create a simple preview interface
+def preview_interface():
+    with gr.Blocks() as preview_interface:
+        gr.Markdown("# Preview Interface")
+        preview_image = gr.Image(source="upload", type="pil", label="Preview your image")
+        gr.Markdown("## Current Image")
+        gr.Image(value="output.png", label="Current Output Image")  # Replace with dynamic update if needed
+        
+    return preview_interface
+
+# Launch the main interface and the additional draw and preview pages
 if __name__ == "__main__":
-    demo.queue().launch(debug=True, share=True)
+    # Launch the main interface with Gradio
+    demo.queue().launch(debug=True)
+
+    # Launch the drawing and preview interfaces
+    draw_interface().launch()
+    preview_interface().launch()
+