Update app.py

app.py

@@ -126,8 +126,14 @@ def initialize_flux_api():
     global FLUX_API_ENABLED, flux_api_client
     
     try:
-        logger.info("Connecting to FLUX API...")
+        logger.info(f"Connecting to FLUX API at {FLUX_API_URL}...")
         flux_api_client = Client(FLUX_API_URL)
+        
+        # API 정보 확인
+        if hasattr(flux_api_client, 'view_api'):
+            api_info = flux_api_client.view_api()
+            logger.info(f"FLUX API endpoints: {api_info}")
+        
         FLUX_API_ENABLED = True
         logger.info("FLUX API connected successfully")
         return True
@@ -667,8 +673,7 @@ def generate_flux_prompt(title: str, content: str, style_key: str) -> str:
     """
     Build a FLUX image-generation prompt for one slide, using six pre-defined
     visual styles (Product Design, Mindmap, Mockup, Infographic, Diagram,
-    Flowchart).  `content` is the raw bullet-point block; ‘+’/‘-’/‘•’/‘●’ 등
-    모든 글머리 기호를 처리한다.
+    Flowchart).
     """
     # 1) clean bullet points (max 8)
     bullets = [
@@ -678,28 +683,33 @@ def generate_flux_prompt(title: str, content: str, style_key: str) -> str:
     ][:8]
 
     if bullets:
-        nodes_block = '\n'.join(f"- {b}" for b in bullets)
         tree_block = '\n'.join(
             f"{'├──' if i < len(bullets)-1 else '└──'} {b}"
             for i, b in enumerate(bullets)
         )
     else:
-        nodes_block = "- (no explicit bullet points) -"
-        tree_block = nodes_block
+        # 불릿이 없으면 제목을 사용
+        tree_block = f"└── {title}"
 
-    # 2) choose template & inject nodes
+    # 2) choose template & inject tree only (not nodes)
     template = EXAMPLE_PROMPTS.get(style_key, EXAMPLE_PROMPTS["Diagram"])
-    prompt_body = template.format(nodes=nodes_block, tree=tree_block)
+    prompt_body = template.format(tree=tree_block)  # ← nodes 제거, tree만 사용
 
-    # 3) stylistic tail (≤120 words total)
+    # 3) stylistic tail
     tail = (
         "corporate colour palette, white background, hand-drawn line style; "
         "boxes, arrows, connectors clearly visible; high-resolution vector."
     )    
 
-    return f"{prompt_body}\n\n{tail}".strip()
-
+    final_prompt = f"{prompt_body}\n\n{tail}".strip()
+    
+    # 디버깅용 로그 추가
+    logger.info(f"[FLUX Prompt] Style: {style_key}")
+    logger.info(f"[FLUX Prompt] Generated: {final_prompt[:200]}...")
+    
+    return final_prompt
 
+ 
 
 def _save_temp_image(img_bytes: bytes, suffix=".png") -> str:
     tmp = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
@@ -2260,7 +2270,9 @@ def generate_ppt_content(topic: str, num_slides: int, additional_context: str, u
 5. 각 섹션 사이에 빈 줄 없음
 6. 총 {num_slides}장 작성
 7. 각 포인트는 '-' 기호로 시작하세요 (이모지는 자동으로 추가됩니다)
-8. 노트는 해당 슬라이드의 내용을 발표자가 청중에게 설명하는 구어체 대본으로 작성하세요"""
+8. 노트는 해당 슬라이드의 내용을 발표자가 청중에게 설명하는 구어체 대본으로 작성하세요
+9. 각 글머리 기호 포인트는 구체적이고 명확하게 작성하세요
+10. 다이어그램이나 플로우차트로 표현하기 좋은 내용은 구조화된 방식으로 작성하세요"""
     else:
         system_prompt = f"""You are a professional PPT presentation expert.
 Create content for exactly {num_slides} slides on the given topic.
@@ -2289,6 +2301,8 @@ Notes: [Speaker script]
 
 (Continue this way until Slide {num_slides})
 
+{layout_instructions}
+
 **Important instructions:**
 1. Each slide starts with "Slide number"
 2. Title: followed by the actual title (no numbers)
@@ -2297,7 +2311,9 @@ Notes: [Speaker script]
 5. No empty lines between sections
 6. Create exactly {num_slides} slides
 7. Start each point with '-' (emojis will be added automatically)
-8. Notes should be a speaker script explaining the slide content in conversational style"""
+8. Notes should be a speaker script explaining the slide content in conversational style
+9. Make each bullet point specific and clear
+10. Structure content that would work well as diagrams or flowcharts"""
 
     # Add search results if web search is performed
     if additional_context:
@@ -2361,9 +2377,21 @@ Notes: [Speaker script]
                     except json.JSONDecodeError:
                         logger.warning(f"JSON parsing failed: {data_str}")
                         continue
+        
+        # 최종 응답 로깅 (디버깅용)
+        logger.info(f"Generated {len(full_response)} characters of content")
+        logger.debug(f"First 500 chars of response: {full_response[:500]}")
                         
+    except requests.exceptions.Timeout:
+        logger.error("LLM API timeout")
+        yield "⚠️ Error: Request timed out. Please try again."
+    except requests.exceptions.RequestException as e:
+        logger.error(f"LLM API request error: {str(e)}")
+        yield f"⚠️ Error: Network error - {str(e)}"
     except Exception as e:
         logger.error(f"LLM API error: {str(e)}")
+        import traceback
+        logger.error(f"Traceback: {traceback.format_exc()}")
         yield f"⚠️ Error generating content: {str(e)}"
 
 ##############################################################################