Create app.py

app.py

@@ -0,0 +1,189 @@
+import gradio as gr
+import requests
+import json
+import os
+
+# 환경변수에서 API 키 로드
+FIREWORKS_API_KEY = os.getenv("FIREWORKS_API_KEY", "")
+
+def enhance_prompt(prompt: str, api_key: str) -> str:
+    """
+    Fireworks AI LLM API를 사용하여 프롬프트를 증강합니다.
+    """
+    if not prompt.strip():
+        return "❌ 프롬프트를 입력해주세요."
+    
+    if not api_key.strip():
+        return "❌ Fireworks API 키를 입력해주세요."
+    
+    system_message = """You are a professional prompt engineer specializing in AI image and video generation. 
+Your task is to enhance user prompts to create more detailed, vivid, and effective prompts for AI image/video generation models.
+
+When enhancing prompts:
+1. Add specific visual details (lighting, composition, style, mood)
+2. Include artistic references or styles when appropriate
+3. Specify camera angles, perspectives, and depth of field
+4. Add color palette suggestions
+5. Include quality modifiers (8K, cinematic, professional, etc.)
+6. Maintain the original intent while enriching the description
+7. Output the enhanced prompt in English for best AI model compatibility
+
+Respond ONLY with the enhanced prompt, no explanations or additional text."""
+
+    url = "https://api.fireworks.ai/inference/v1/chat/completions"
+    
+    payload = {
+        "model": "accounts/fireworks/models/gpt-oss-120b",
+        "max_tokens": 4096,
+        "top_p": 1,
+        "top_k": 40,
+        "presence_penalty": 0,
+        "frequency_penalty": 0,
+        "temperature": 0.6,
+        "messages": [
+            {
+                "role": "system",
+                "content": system_message
+            },
+            {
+                "role": "user",
+                "content": f"다음 프롬프트를 이미지/영상 생성 AI에 최적화된 상세하고 풍부한 프롬프트로 증강해주세요:\n\n\"{prompt}\""
+            }
+        ]
+    }
+    
+    headers = {
+        "Accept": "application/json",
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    }
+    
+    try:
+        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=60)
+        response.raise_for_status()
+        
+        data = response.json()
+        enhanced = data.get("choices", [{}])[0].get("message", {}).get("content", "")
+        
+        if enhanced:
+            return enhanced
+        else:
+            return "❌ 응답에서 증강된 프롬프트를 찾을 수 없습니다."
+            
+    except requests.exceptions.Timeout:
+        return "❌ API 요청 시간 초과. 다시 시도해주세요."
+    except requests.exceptions.HTTPError as e:
+        return f"❌ API 오류: {e.response.status_code} - {e.response.text}"
+    except requests.exceptions.RequestException as e:
+        return f"❌ 네트워크 오류: {str(e)}"
+    except Exception as e:
+        return f"❌ 오류 발생: {str(e)}"
+
+
+# Example prompts
+examples = [
+    ["한복을 입은 여성이 전통 한옥 마당에서 부채를 들고 있다"],
+    ["사이버펑크 도시의 네온사인이 빛나는 밤거리"],
+    ["바다 위 일몰과 작은 돛단배"],
+    ["눈 덮인 산속 작은 오두막에서 연기가 피어오른다"],
+    ["미래 도시의 하늘을 나는 자동차들"],
+]
+
+# Build the Gradio interface
+with gr.Blocks(title="AI 프롬프트 증강기", theme=gr.themes.Soft(primary_hue="teal", secondary_hue="cyan")) as demo:
+    
+    gr.HTML("""
+        <div style="text-align: center; padding: 20px 0;">
+            <div style="display: inline-flex; align-items: center; gap: 12px; margin-bottom: 10px;">
+                <div style="
+                    width: 50px; 
+                    height: 50px; 
+                    border-radius: 12px; 
+                    background: linear-gradient(135deg, #4ecdc4 0%, #45b7d1 100%);
+                    display: flex;
+                    align-items: center;
+                    justify-content: center;
+                    font-size: 24px;
+                ">✨</div>
+                <h1 style="
+                    font-size: 2.5rem;
+                    font-weight: 800;
+                    background: linear-gradient(135deg, #4ecdc4 0%, #45b7d1 50%, #96e6a1 100%);
+                    -webkit-background-clip: text;
+                    -webkit-text-fill-color: transparent;
+                    margin: 0;
+                ">AI 프롬프트 증강기</h1>
+            </div>
+            <p style="color: #6b7280; font-size: 1.1rem; margin: 0;">
+                이미지 & 영상 생성을 위한 프롬프트를 AI가 자동으로 증강합니다
+            </p>
+        </div>
+    """)
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            api_key_input = gr.Textbox(
+                label="🔑 Fireworks API Key",
+                placeholder="fw_...",
+                type="password",
+                value=FIREWORKS_API_KEY,
+                info="Fireworks AI API 키를 입력하세요 (환경변수 FIREWORKS_API_KEY로도 설정 가능)"
+            )
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            prompt_input = gr.Textbox(
+                label="📝 원본 프롬프트",
+                placeholder="이미지나 영상으로 생성하고 싶은 장면을 설명해주세요...\n예: 한복을 입은 여성이 벚꽃 아래에서 걷고 있다",
+                lines=5,
+                max_lines=10
+            )
+            
+            enhance_btn = gr.Button(
+                "✨ 프롬프트 증강하기",
+                variant="primary",
+                size="lg"
+            )
+            
+            gr.Examples(
+                examples=examples,
+                inputs=prompt_input,
+                label="💡 예시 프롬프트"
+            )
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            output_text = gr.Textbox(
+                label="🚀 증강된 프롬프트 (복사하여 사용하세요)",
+                placeholder="증강된 프롬프트가 여기에 표시됩니다...",
+                lines=8,
+                max_lines=15,
+                interactive=True
+            )
+    
+    gr.HTML("""
+        <div style="text-align: center; padding: 20px 0; color: #6b7280; font-size: 0.9rem;">
+            <p>💡 <strong>팁:</strong> 증강된 프롬프트는 영어로 출력되어 대부분의 AI 이미지/영상 생성 모델과 호환됩니다.</p>
+            <p style="margin-top: 8px;">Powered by Fireworks AI (gpt-oss-120b) | Built with Gradio</p>
+        </div>
+    """)
+    
+    # Event handlers
+    enhance_btn.click(
+        fn=enhance_prompt,
+        inputs=[prompt_input, api_key_input],
+        outputs=output_text,
+        show_progress=True
+    )
+    
+    # Allow Enter key to submit (Shift+Enter for new line)
+    prompt_input.submit(
+        fn=enhance_prompt,
+        inputs=[prompt_input, api_key_input],
+        outputs=output_text,
+        show_progress=True
+    )
+
+
+if __name__ == "__main__":
+    demo.launch()