Update app.py

app.py

@@ -16,6 +16,7 @@ import uuid
 import json
 from huggingface_hub.utils._errors import HfHubHTTPError
 from gradio_client import Client
+from urllib.parse import urljoin
 
 def setup_session():
     try:
@@ -589,8 +590,7 @@ def process_all_titles(category, style, topic):
     return "\n".join(results)
 
 
-# 이미지 생성 API 클라이언트 설정
-image_client = Client("http://hugpu.ai:7899/")
+BASE_URL = "http://hugpu.ai:7899"  # 실제 서버 주소로 변경하세요
 
 def generate_image(prompt):
     try:
@@ -598,9 +598,15 @@ def generate_image(prompt):
             prompt,
             api_name="/generate"
         )
-        # API 응답에서 이미지 URL을 추출합니다.
-        # 실제 응답 구조에 따라 이 부분을 조정해야 할 수 있습니다.
-        image_url = result[0] if isinstance(result, list) else result
+        # API 응답에서 이미지 경로를 추출합니다.
+        image_path = result[0] if isinstance(result, list) else result
+        
+        # 로컬 파일 경로를 웹 URL로 변환
+        if image_path.startswith('/'):
+            image_url = urljoin(BASE_URL, image_path)
+        else:
+            image_url = image_path
+        
         return image_url
     except Exception as e:
         print(f"이미지 생성 중 오류 발생: {str(e)}")
@@ -622,15 +628,18 @@ def process_all_titles(category, style, topic, num_titles):
 
             # 이미지 생성
             image_url = generate_image(title)
-            
+
             if image_url:
                 # 이미지 URL을 블로그 내용 끝에 추가
-                blog_content += f"\n\n[생성된 이미지]({image_url})"
+                blog_content += f"\n\n![생성된 이미지]({image_url})"
+            else:
+                blog_content += "\n\n이미지 생성에 실패했습니다."
 
             # 포스팅 전송
             send_result = send_to_blogger(title, blog_content)
-            
+        
             results.append(f"제목: {title}\n전송 결과: {send_result}\n생성된 이미지 URL: {image_url}\n\n")
+
         except Exception as e:
             results.append(f"제목: {title}\n처리 중 오류 발생: {str(e)}\n\n")