Abdulrahman1989 commited on
Commit
17397c2
·
1 Parent(s): 30f0bac

Add share=True for for public link

Browse files
Files changed (1) hide show
  1. app.py +14 -15
app.py CHANGED
@@ -4,27 +4,27 @@ import os
4
 
5
  class SDXLGenerator:
6
  def generate_image(self, prompt):
7
- # Placeholder for image generation (e.g., returning a sample image or empty image)
8
  return "Placeholder for Generated Image"
9
 
10
-
11
  class ControlNetProcessor:
12
  def controlnet_image(self, image):
13
- # Placeholder for ControlNet processing (e.g., returning a sample image or empty image)
14
  return "Placeholder for ControlNet Output Image"
15
 
16
-
17
  class VideoGenerator:
18
  def generate_3d_video(self, controlled_image):
19
- # Placeholder: creating a dummy video for demonstration purposes.
20
- video_path = "generated_video.mp4"
21
  with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmp:
22
- # Placeholder: creating a dummy video for demonstration purposes.
23
- os.system(f"ffmpeg -f lavfi -i color=c=blue:s=320x240:d=5 -vf drawtext=fontfile=/path/to/font.ttf:text='3D Model':fontsize=24:fontcolor=white:x=(w-text_w)/2:y=(h-text_h)/2 {tmp.name}")
 
 
 
 
24
  video_path = tmp.name
25
  return video_path
26
 
27
-
28
  class GradioApp:
29
  def __init__(self):
30
  self.sdxl_generator = SDXLGenerator()
@@ -40,18 +40,17 @@ class GradioApp:
40
  def launch(self):
41
  interface = gr.Interface(
42
  fn=self.full_pipeline,
43
- inputs=[gr.Textbox(label="Input Prompt")],
44
  outputs=[
45
- gr.Image(type="pil", label="Generated Image"),
46
- gr.Image(type="pil", label="ControlNet Output Image"),
47
  gr.Video(label="3D Model Video")
48
  ],
49
  title="SDXL to ControlNet to 3D Pipeline",
50
  description="Generate an image using SDXL, refine it with ControlNet, and generate a 3D video output."
51
  )
52
- interface.launch()
53
-
54
 
55
  if __name__ == "__main__":
56
  app = GradioApp()
57
- app.launch()
 
4
 
5
  class SDXLGenerator:
6
  def generate_image(self, prompt):
7
+ # Placeholder for image generation (e.g., returning a sample image or placeholder text)
8
  return "Placeholder for Generated Image"
9
 
 
10
  class ControlNetProcessor:
11
  def controlnet_image(self, image):
12
+ # Placeholder for ControlNet processing (e.g., returning a processed image or placeholder text)
13
  return "Placeholder for ControlNet Output Image"
14
 
 
15
  class VideoGenerator:
16
  def generate_3d_video(self, controlled_image):
17
+ # Creating a temporary video with a placeholder for demonstration purposes.
 
18
  with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmp:
19
+ # Generates a sample video with FFmpeg using a solid color and overlay text
20
+ os.system(
21
+ f"ffmpeg -f lavfi -i color=c=blue:s=320x240:d=5 "
22
+ f"-vf drawtext=fontfile=/path/to/font.ttf:text='3D Model':fontsize=24:fontcolor=white:x=(w-text_w)/2:y=(h-text_h)/2 "
23
+ f"{tmp.name}"
24
+ )
25
  video_path = tmp.name
26
  return video_path
27
 
 
28
  class GradioApp:
29
  def __init__(self):
30
  self.sdxl_generator = SDXLGenerator()
 
40
  def launch(self):
41
  interface = gr.Interface(
42
  fn=self.full_pipeline,
43
+ inputs=gr.Textbox(label="Input Prompt"),
44
  outputs=[
45
+ gr.Textbox(label="Generated Image Placeholder"),
46
+ gr.Textbox(label="ControlNet Output Image Placeholder"),
47
  gr.Video(label="3D Model Video")
48
  ],
49
  title="SDXL to ControlNet to 3D Pipeline",
50
  description="Generate an image using SDXL, refine it with ControlNet, and generate a 3D video output."
51
  )
52
+ interface.launch(share=True) # Added `share=True` for public link
 
53
 
54
  if __name__ == "__main__":
55
  app = GradioApp()
56
+ app.launch()