Create app.py

app.py

@@ -0,0 +1,143 @@
+import gradio as gr
+with gr.Blocks(theme=gr.themes.Default()) as app:
+    with gr.Tab("삽화 생성"):
+        with gr.Row():
+            # 1
+            gr.Markdown(
+            value="""
+            # 삽화 생성
+            짧은 소설에 어울리는 이미지를 생성할 수 있습니다.
+            """
+            )
+        with gr.Row():
+            # column1
+            with gr.Column(scale=5):
+                # 2
+                pos_prompt = gr.Textbox(
+                    label="이미지 생성 프롬프트를 작성해 주세요",
+                    value="ultra realistic close up portrait ((beautiful pale cyberpunk female with heavy black eyeliner))",
+                    lines=8,
+                    interactive=True,
+                )
+                with gr.Row():
+                    # 3
+                    auto_prompt_generator = gr.Textbox(
+                        label="자동 프롬프트 생성",
+                        lines=6,
+                        placeholder="이미지 생성을 위한 소설 내용을 작성해 주세요.\n자동으로 프롬프트가 생성됩니다.",
+                        scale=7,
+                    )
+                    # 4
+                    prompt_generator_btn = gr.Button(scale=1, value="자동\n생성")
+
+                with gr.Group():
+                    with gr.Row():
+                        # 5
+                        img_width = gr.Slider(
+                            label="=Width", maximum=1024, value=512, interactive=True
+                        )
+                        # 6
+                        img_height = gr.Slider(
+                            label="=Height", maximum=1024, value=512, interactive=True
+                        )
+
+            # column2
+            with gr.Column(scale=3):
+                # 7
+                output_status = gr.Textbox(
+                    show_label=False, lines=1, placeholder="이미지 상태가 출력됩니다."
+                )
+                # 8
+                generator_img = gr.Image(
+                    value="https://pub-3626123a908346a7a8be8d9295f44e26.r2.dev/generations/8c595b57-563c-4417-9bfb-96aaebbb30b3-0.png",
+                    label="이미지가 생성됩니다.",
+                )
+                # 9
+                generator_img_btn = gr.Button(value="이미지 생성")
+                # 10
+                refresh_img_btn = gr.Button(value="이미지 새로고침")
+        # 자동 생성 버튼 클릭
+        prompt_generator_btn.click(
+            fn=novel_keyword,
+            inputs=[auto_prompt_generator],
+            outputs=[pos_prompt]
+        ) # 이미지 생성 버튼 클릭
+        generator_img_btn.click(
+            fn=sd_call,
+            inputs=[pos_prompt, img_width, img_height],
+            outputs=[generator_img, output_status]
+        ) # 이미지 새로고침 버튼 클릭
+        refresh_img_btn.click(
+            fn=sd_recall,
+            inputs=[output_status],
+            outputs=[generator_img]
+        )
+    with gr.Tab("이미지 편집") as edit_tab:
+        with gr.Row():
+            #1
+            gr.Markdown(
+            value="""
+            # 이미지 편집
+            생성한 이미지를 편집할 수 있습니다.
+            """)
+        with gr.Row():
+            #2
+            edit_prompt = gr.Textbox(
+                label="이미지 수정 프롬프트를 작성해 주세요",
+                value="black hair",
+                lines=5,
+                interactive=True,
+                scale=7
+            )
+            #3
+            edit_btn = gr.Button(
+                value="이미지 편집",
+                scale=1
+            )
+        with gr.Row():
+            #4
+            edit_status = gr.Textbox(
+                show_label=False,
+                lines=1,
+                placeholder="편집 이미지 상태가 출력됩니다.",
+                scale=7
+            )
+            #5
+            refresh_edit_btn = gr.Button(
+                value="이미지 새로고침",
+                scale=1
+            )
+        with gr.Row():
+            #6
+            org_img = gr.ImageMask(
+                label="수정 전 이미지",
+                image_mode='RGB',
+                brush = gr.Brush(
+                    default_size=20,
+                    colors=["#FFFFFF"],
+                    color_mode="fixed",
+                    ),
+                show_label=True
+                )
+            #7
+            edit_img = gr.Image(
+                label="수정 후 이미지",
+            )
+    edit_tab.select(
+        fn=edit_load_img,
+        inputs=[generator_img],
+        outputs=[org_img]
+    )
+    #이미지 편집 버튼 클릭
+    edit_btn.click(
+        fn=edit_img_generator,
+        inputs=[org_img, edit_prompt],
+        outputs=[edit_img, edit_status]
+    )
+    #이미지
+    refresh_edit_btn.click(
+        fn=sd_edit_recall,
+        inputs=[edit_status],
+        outputs=[edit_img]
+    )
+app.launch(debug=True)