Update app.py

app.py

@@ -309,39 +309,27 @@ footer {display: none}
 }
 """
 
-try:
-    prompt_enhancer = pipeline(
-        "text-generation", 
-        model="CohereForAI/c4ai-command-r-plus-08-2024",
-        token=HF_TOKEN
-    )
-except Exception as e:
-    print(f"Error initializing prompt enhancer: {str(e)}")
-    prompt_enhancer = None
+import requests
 
 def enhance_prompt(prompt: str) -> str:
     """프롬프트를 애니메이션 스타일로 증강"""
     try:
-        if prompt_enhancer is None:
-            return prompt
-
-        base_prompt = f"Convert this into a detailed anime-style image prompt: {prompt}"
-        messages = [{"role": "user", "content": base_prompt}]
-        
-        response = prompt_enhancer(messages)
-        enhanced = response[0]['generated_text']
-        
         # 기본 품질 향상 프롬프트 추가
         enhancements = [
             "masterpiece, best quality, highly detailed",
             "anime style, animation style",
             "vibrant colors, perfect lighting",
             "professional composition",
+            "dynamic pose, expressive features",
+            "detailed background, perfect shadows",
             "[trigger]"
         ]
         
+        # 애니메이션 스타일 프롬프트 변환
+        anime_style_prompt = f"an animated {prompt}, detailed anime art style"
+        
         # 최종 프롬프트 구성
-        final_prompt = f"{enhanced}, {', '.join(enhancements)}"
+        final_prompt = f"{anime_style_prompt}, {', '.join(enhancements)}"
         print(f"Enhanced prompt: {final_prompt}")
         
         return final_prompt
@@ -349,6 +337,14 @@ def enhance_prompt(prompt: str) -> str:
         print(f"Prompt enhancement failed: {str(e)}")
         return prompt
 
+# 기존의 pipeline 초기화 부분 제거
+# try:
+#     prompt_enhancer = pipeline(...)
+# except Exception as e:
+#     print(f"Error initializing prompt enhancer: {str(e)}")
+#     prompt_enhancer = None
+
+
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">