Update app.py

app.py

@@ -317,6 +317,22 @@ def inference(
         error_img = Image.new('RGB', (width, height), color='red')
         return error_img, seed, load_generated_images()
 
+# 프롬프트 내 텍스트를 추출하여 텍스트 렌더링 입력란에 자동으로 채우는 함수
+def extract_text_for_rendering(prompt):
+    """
+    프롬프트에서 따옴표('' 또는 "")로 둘러싸인 텍스트를 찾아
+    텍스트 렌더링 입력란에 자동으로 채우기 위한 함수
+    """
+    # 따옴표로 둘러싸인 텍스트 찾기
+    single_quotes = re.findall(r"'([^']*)'", prompt)
+    double_quotes = re.findall(r'"([^"]*)"', prompt)
+    
+    # 추출된 텍스트가 있으면 첫 번째 것을 반환
+    extracted_texts = single_quotes + double_quotes
+    if extracted_texts:
+        return extracted_texts[0]
+    return ""
+
 ###############################################################################
 # 6. Gradio UI
 ###############################################################################
@@ -639,7 +655,7 @@ with gr.Blocks(css=css, analytics_enabled=False, theme="soft") as demo:
         return load_generated_images()
 
     def clear_output():
-        return "", gr.update(value=None), seed, '<div class="status-complete">Ready to generate</div>'
+        return "", gr.update(value=None), seed, '<div class="status-complete">Ready to generate</div>', ""
     
     def before_generate():
         return '<div class="status-generating">Generating image...</div>'
@@ -659,17 +675,22 @@ with gr.Blocks(css=css, analytics_enabled=False, theme="soft") as demo:
     clear_button.click(
         fn=clear_output,
         inputs=None,
-        outputs=[prompt, result, seed_text, generation_status]
+        outputs=[prompt, result, seed_text, generation_status, text_rendering]
     )
     
     # 1) 상태 표시
-    # 2) 이미지 생성
-    # 3) 상태 업데이트
-    # 4) 텍스트 렌더링(있다면)
+    # 2) 텍스트 추출
+    # 3) 이미지 생성
+    # 4) 상태 업데이트
+    # 5) 텍스트 렌더링(있다면)
     run_button.click(
         fn=before_generate,
         inputs=None,
         outputs=generation_status,
+    ).then(
+        fn=extract_text_for_rendering,  # 추가: 텍스트 추출 함수
+        inputs=prompt,
+        outputs=text_rendering,  # 추출된 텍스트로 텍스트 렌더링 필드 업데이트
     ).then(
         fn=inference,
         inputs=[
@@ -698,6 +719,10 @@ with gr.Blocks(css=css, analytics_enabled=False, theme="soft") as demo:
         fn=before_generate,
         inputs=None,
         outputs=generation_status,
+    ).then(
+        fn=extract_text_for_rendering,  # 추가: 텍스트 추출 함수
+        inputs=prompt,
+        outputs=text_rendering,  # 추출된 텍스트로 텍스트 렌더링 필드 업데이트
     ).then(
         fn=inference,
         inputs=[
@@ -749,4 +774,4 @@ try:
 except Exception as e:
     logging.error(f"Error during launch: {e}")
     logging.info("Trying alternative launch configuration...")
-    demo.launch(debug=True, show_api=False, share=False)
+    demo.launch(debug=True, show_api=False, share=False)