initial commit

app.py

@@ -0,0 +1,158 @@
+import gradio as gr
+import time
+from PIL import Image
+
+
+# Initialize a counter to keep track of the number of times the function is called
+call_counter = 0
+
+# Function to display different images based on the call count
+def generate_image(text):
+    global call_counter
+
+    # Define image paths for the first and second calls
+    image_paths = ["/content/yc01.jpg", "/content/yc02.png"]
+
+    # Check if the call_counter is within the range of available image paths
+    if call_counter < len(image_paths):
+        image_path = image_paths[call_counter]
+        call_counter += 1
+    else:
+        image_path = None
+
+    # If the image path exists, return it; otherwise, return None
+    if image_path:
+        image = Image.open(image_path)
+        return image
+    else:
+        return None
+
+
+# Function to show thumbnail
+def get_ytmetas(youtube_link):
+
+    # Define image paths for the first and second calls
+    image_path = ["/content/playlist.png"]
+
+    # If the image path exists, return it; otherwise, return None
+    if image_path:
+        image = Image.open(image_path)
+        return f"Dalton & Michael", f"Dalton Caldwell and Michael Seibel talk startups, Silicon Valley and advice for founders."
+        #return image, "Dalton & Michael", "Dalton Caldwell and Michael Seibel talk startups, Silicon Valley and advice for founders."
+    else:
+        return None
+
+# Function to fine-tune a model
+def train_model(youtube_link):
+    time.sleep(1)
+    return "Model successfully trained for Datlon & Michael."
+
+
+with gr.Blocks(title = "Old Fat Boy") as demo:
+    with gr.Tab("Train"):
+            with gr.Row():
+                tb_youtubelink = gr.Textbox(label="Youtube Link" )
+            with gr.Row():
+                btn_run = gr.Button("Train",variant="primary")
+            with gr.Row():
+                tb_indicator = gr.Textbox(label="Output")
+
+            btn_run.click(fn=train_model,inputs=[tb_youtubelink],outputs=[tb_indicator])
+
+    with gr.Tab("Generate"):
+        with gr.Row():
+            prompt = gr.Textbox(label="Prompt" )
+            image_output = gr.Image(label="Image")
+        image_button = gr.Button("Generate Image")
+
+    image_button.click(generate_image, inputs=prompt, outputs=image_output)
+
+    with gr.Tab("Edit"):
+            with gr.Column():
+              edit_image = gr.Image(label="Generated Image")
+                # Left --> tools
+            with gr.Row():
+              # Element 1
+              with gr.Column():
+                element1 = gr.Image(label="Element1")
+                # Drag
+                with gr.Row():
+                    with gr.Column(scale=1, min_width=10):
+                        gr.Markdown(value="Drag", show_label=False)
+                    with gr.Column(scale=4, min_width=10):
+                        with gr.Row():
+                            with gr.Column(scale=1, min_width=10):
+                                enable_add_points = gr.Button("Add Points")
+                            with gr.Column(scale=1, min_width=10):
+                                undo_points = gr.Button("Reset Points")
+                        with gr.Row():
+                            with gr.Column(scale=1, min_width=10):
+                                form_start_btn = gr.Button("Start")
+                            with gr.Column(scale=1, min_width=10):
+                                form_stop_btn = gr.Button("Apply")
+
+
+                form_draw_interval_number = gr.Number(
+                    #value=global_state.value["draw_interval"],
+                    label="Draw Interval (steps)",
+                    interactive=True,
+                    visible=False,
+                )
+
+              # Element 2
+              with gr.Column():
+                element1 = gr.Image(label="Element2")
+                # Drag
+                with gr.Row():
+                    with gr.Column(scale=1, min_width=10):
+                        gr.Markdown(value="Drag", show_label=False)
+                    with gr.Column(scale=4, min_width=10):
+                        with gr.Row():
+                            with gr.Column(scale=1, min_width=10):
+                                enable_add_points = gr.Button("Add Points")
+                            with gr.Column(scale=1, min_width=10):
+                                undo_points = gr.Button("Reset Points")
+                        with gr.Row():
+                            with gr.Column(scale=1, min_width=10):
+                                form_start_btn = gr.Button("Start")
+                            with gr.Column(scale=1, min_width=10):
+                                form_stop_btn = gr.Button("Apply")
+
+
+                form_draw_interval_number = gr.Number(
+                    #value=global_state.value["draw_interval"],
+                    label="Draw Interval (steps)",
+                    interactive=True,
+                    visible=False,
+                )
+
+              # Element 3
+              with gr.Column():
+                element1 = gr.Image(label="Element3")
+                # Drag
+                with gr.Row():
+                    with gr.Column(scale=1, min_width=10):
+                        gr.Markdown(value="Drag", show_label=False)
+                    with gr.Column(scale=4, min_width=10):
+                        with gr.Row():
+                            with gr.Column(scale=1, min_width=10):
+                                enable_add_points = gr.Button("Add Points")
+                            with gr.Column(scale=1, min_width=10):
+                                undo_points = gr.Button("Reset Points")
+                        with gr.Row():
+                            with gr.Column(scale=1, min_width=10):
+                                form_start_btn = gr.Button("Start")
+                            with gr.Column(scale=1, min_width=10):
+                                form_stop_btn = gr.Button("Apply")
+
+                form_draw_interval_number = gr.Number(
+                    #value=global_state.value["draw_interval"],
+                    label="Draw Interval (steps)",
+                    interactive=True,
+                    visible=False,
+                )
+
+
+if __name__ == "__main__":
+    demo.launch(share=False, debug=True, favicon_path='oldfatboy.png')
+