Update app.py

app.py

@@ -15,6 +15,7 @@ from diffusers import StableDiffusionXLPipeline
 import uuid
 import json
 from huggingface_hub.utils._errors import HfHubHTTPError
+from gradio_client import Client
 
 def setup_session():
     try:
@@ -588,24 +589,24 @@ def process_all_titles(category, style, topic):
     return "\n".join(results)
 
 
-# 번역 모델 설정
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
-
-# 이미지 생성 모델 설정
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium", torch_dtype=torch.float16)
-pipe = pipe.to(device)
-
-def translate_title(title):
-    translated = translator(title, max_length=128)[0]['translation_text']
-    return translated
+# 이미지 생성 API 클라이언트 설정
+image_client = Client("http://hugpu.ai:7899/")
 
 def generate_image(prompt):
-    image = pipe(prompt).images[0]
-    image_path = f"generated_image_{uuid.uuid4()}.png"
-    image.save(image_path)
-    return image_path
+    try:
+        result = image_client.predict(
+            prompt,
+            api_name="/generate"
+        )
+        # API 응답에서 이미지 URL을 추출합니다.
+        # 실제 응답 구조에 따라 이 부분을 조정해야 할 수 있습니다.
+        image_url = result[0] if isinstance(result, list) else result
+        return image_url
+    except Exception as e:
+        print(f"이미지 생성 중 오류 발생: {str(e)}")
+        return None
 
+# process_all_titles 함수 수정
 def process_all_titles(category, style, topic, num_titles):
     title_suggestions, _ = suggest_title(category, style, topic, "", "", "")
     titles = title_suggestions.split('\n')
@@ -619,17 +620,17 @@ def process_all_titles(category, style, topic, num_titles):
                 results.append(f"제목: {title}\n생성 실패: {blog_content}\n\n")
                 continue
 
-            # 제목 번역 및 이미지 생성
-            translated_title = translate_title(title)
-            image_path = generate_image(translated_title)
+            # 이미지 생성
+            image_url = generate_image(title)
             
-            # 이미지 링크를 블로그 내용 끝에 추가
-            blog_content += f"\n\n[생성된 이미지]({image_path})"
+            if image_url:
+                # 이미지 URL을 블로그 내용 끝에 추가
+                blog_content += f"\n\n[생성된 이미지]({image_url})"
 
             # 포스팅 전송
             send_result = send_to_blogger(title, blog_content)
             
-            results.append(f"제목: {title}\n전송 결과: {send_result}\n생성된 이미지: {image_path}\n\n")
+            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")
         
@@ -637,7 +638,6 @@ def process_all_titles(category, style, topic, num_titles):
     
     return "\n".join(results)
 
-
 with gr.Blocks() as demo:
     gr.Markdown(f"# {title}")
     
@@ -656,12 +656,11 @@ with gr.Blocks() as demo:
 
     start_btn = gr.Button("시작")
     result_output = gr.Textbox(label="처리 결과", lines=20)
-    generated_image = gr.Image(label="생성된 이미지")
 
     start_btn.click(
         fn=process_all_titles,
         inputs=[category, style, topic, num_titles],
-        outputs=[result_output, generated_image]
+        outputs=[result_output]
     )
 
     category.change(fn=update_prompts_and_description, inputs=[category, style], outputs=[style_description])