Update app.py

app.py

@@ -1139,7 +1139,8 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
     # AI 번역 탭 추가
         with gr.Tab("AI 기사 생성"):
             gr.Markdown("뉴스 URL을 입력하면 AI가 한국어로 번역하여 기사 형식으로 작성합니다.")
-        
+            gr.Markdown("이미지 생성: https://huggingface.co/spaces/ginipick/FLUXllama ")
+            
             with gr.Column():
                 chatbot = gr.Chatbot(height=600)
             
@@ -1148,32 +1149,46 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                         label="뉴스 URL",
                         placeholder="https://..."
                     )
-            
+
                 with gr.Accordion("고급 설정", open=False):
+
                     system_message = gr.Textbox(
-                        value="""You are a professional translator and journalist. Follow these steps strictly:
-                        1. TRANSLATION
-                        - Start with ===번역=== marker
-                        - Provide accurate Korean translation
-                        - Maintain original meaning and context
-    
-                        2. ARTICLE WRITING
-                        - Start with ===기사=== marker
-                        - Write a new Korean news article based on the translation
-                        - Follow newspaper article format
-                        - Use formal news writing style
-                        - End sentences with '다.'
-                        - Include headline and subheadline
-                        - Organize paragraphs clearly
-                        - Put key information first
-                        - Use quotes appropriately
-        
-                        IMPORTANT:
-                        - Must complete both steps in order
-                        - Clearly separate each section with markers
-                        - Never skip or combine steps""",
-                        label="System message"
-                    )
+                         value="""You are a professional translator and journalist. Follow these steps strictly:
+                         1. TRANSLATION
+                         - Start with ===번역=== marker
+                         - Provide accurate Korean translation
+                         - Maintain original meaning and context
+
+                         2. ARTICLE WRITING
+                         - Start with ===기사=== marker
+                         - Write a new Korean news article based on the translation
+                         - Follow newspaper article format
+                         - Use formal news writing style
+                         - End sentences with '다.'
+                         - Include headline and subheadline
+                         - Organize paragraphs clearly
+                         - Put key information first
+                         - Use quotes appropriately
+                         
+                         3. IMAGE PROMPT GENERATION
+                         - Start with ===이미지 프롬프트=== marker
+                         - Create detailed Korean prompts for image generation
+                         - Prompts should reflect the article's main theme and content
+                         - Include key visual elements mentioned in the article
+                         - Specify style, mood, and composition
+                         - Format: "이미지 설명: [상세 설명]"
+                         - Add style keywords: "스타일: [관련 키워드들]"
+                         - Add mood keywords: "분위기: [관련 키워드들]"
+
+                         IMPORTANT:
+                         - Must complete all three steps in order
+                         - Clearly separate each section with markers
+                         - Never skip or combine steps
+                         - Ensure image prompts align with article content""",
+                         label="System message"
+                     )                    
+                    
+                
 
                     max_tokens = gr.Slider(
                         minimum=1,