Create main.py

main.py

@@ -0,0 +1,147 @@
+import os
+import httpx
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+from pydantic import BaseModel
+from pathlib import Path
+from dotenv import load_dotenv
+
+load_dotenv()
+
+app = FastAPI()
+
+CODESTRAL_API_KEY = os.getenv("CODESTRAL_API_KEY")
+CODESTRAL_FIM_ENDPOINT = "https://codestral.mistral.ai/v1/fim/completions"
+CODESTRAL_CHAT_ENDPOINT = "https://codestral.mistral.ai/v1/chat/completions"
+
+# --- Pydantic 모델 (변경 없음) ---
+class HtmlGenRequest(BaseModel):
+    prompt: str
+
+class CodeGenRequest(BaseModel):
+    prompt: str
+    language: str
+
+class CodeCompleteRequest(BaseModel):
+    prefix: str
+    suffix: str
+    language: str
+
+# --- 경로 설정 (루트 디렉토리 기준) ---
+# main.py와 index.html, style.css, script.js가 모두 같은 레벨(/app)에 위치
+CURRENT_DIR = Path(__file__).resolve().parent # /app
+
+# --- API 엔드포인트 (로직 변경 없음, 이전과 동일) ---
+@app.post("/api/generate-html")
+async def generate_html_endpoint(payload: HtmlGenRequest):
+    if not CODESTRAL_API_KEY:
+        raise HTTPException(status_code=500, detail="API Key not configured on server. Check Hugging Face Space Secrets.")
+    async with httpx.AsyncClient(timeout=60.0) as client:
+        try:
+            response = await client.post(
+                CODESTRAL_CHAT_ENDPOINT,
+                headers={"Authorization": f"Bearer {CODESTRAL_API_KEY}", "Content-Type": "application/json", "Accept": "application/json"},
+                json={"model": "codestral-latest", "messages": [{"role": "system", "content": "You are an expert HTML and web designer..."}, {"role": "user", "content": payload.prompt}], "temperature": 0.3}
+            )
+            response.raise_for_status()
+            data = response.json()
+            html_content = data.get("choices", [{}])[0].get("message", {}).get("content", "").strip() or "<!-- AI HTML generation failed -->"
+            return {"html": html_content}
+        except httpx.HTTPStatusError as e:
+            print(f"Codestral API Error (HTML): {e.response.status_code} - {e.response.text}")
+            raise HTTPException(status_code=e.response.status_code, detail=f"Codestral API error: {e.response.text}")
+        except Exception as e:
+            print(f"Server Error (HTML): {e}")
+            raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/api/generate-code")
+async def generate_code_endpoint(payload: CodeGenRequest):
+    if not CODESTRAL_API_KEY:
+        raise HTTPException(status_code=500, detail="API Key not configured on server.")
+    async with httpx.AsyncClient(timeout=60.0) as client:
+        try:
+            response = await client.post(
+                CODESTRAL_CHAT_ENDPOINT,
+                headers={"Authorization": f"Bearer {CODESTRAL_API_KEY}", "Content-Type": "application/json", "Accept": "application/json"},
+                json={"model": "codestral-latest", "messages": [{"role": "system", "content": f"You are an expert {payload.language} programmer..."}, {"role": "user", "content": payload.prompt}], "temperature": 0.2}
+            )
+            response.raise_for_status()
+            data = response.json()
+            code_content = data.get("choices", [{}])[0].get("message", {}).get("content", "").strip() or f"// AI {payload.language} code generation failed."
+            return {"code": code_content}
+        except httpx.HTTPStatusError as e:
+            print(f"Codestral API Error (Code): {e.response.status_code} - {e.response.text}")
+            raise HTTPException(status_code=e.response.status_code, detail=f"Codestral API error: {e.response.text}")
+        except Exception as e:
+            print(f"Server Error (Code): {e}")
+            raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/api/complete-code")
+async def complete_code_endpoint(payload: CodeCompleteRequest):
+    if not CODESTRAL_API_KEY:
+        raise HTTPException(status_code=500, detail="API Key not configured on server.")
+    if payload.prefix is None or payload.suffix is None:
+         raise HTTPException(status_code=400, detail="Prefix and suffix are required.")
+    async with httpx.AsyncClient(timeout=30.0) as client:
+        try:
+            api_payload = {"model": "codestral-latest", "prompt": payload.prefix, "suffix": payload.suffix, "temperature": 0.1}
+            response = await client.post(
+                CODESTRAL_FIM_ENDPOINT,
+                headers={"Authorization": f"Bearer {CODESTRAL_API_KEY}", "Content-Type": "application/json", "Accept": "application/json"},
+                json=api_payload
+            )
+            response.raise_for_status()
+            data = response.json()
+            completion_text = data.get("choices", [{}])[0].get("text", "")
+            return {"completion": completion_text}
+        except httpx.HTTPStatusError as e:
+            print(f"Codestral API Error (FIM): {e.response.status_code} - {e.response.text}")
+            raise HTTPException(status_code=e.response.status_code, detail=f"Codestral API error: {e.response.text}")
+        except Exception as e:
+            print(f"Server Error (FIM): {e}")
+            raise HTTPException(status_code=500, detail=str(e))
+
+
+# --- 프론트엔드 정적 파일 및 SPA 라우팅 (루트 디렉토리 기준) ---
+
+# style.css와 script.js를 위한 StaticFiles 마운트
+# '/static' 경로로 요청오면 현재 디렉토리의 파일들을 서빙 (예: /static/style.css -> ./style.css)
+# 하지만 HTML에서 <link rel="stylesheet" href="style.css"> 처럼 직접 참조하므로,
+# 아래의 index.html을 서빙하는 라우트와 함께, 개별 파일 서빙 라우트가 필요.
+
+# CSS 파일 서빙
+@app.get("/style.css")
+async def serve_css():
+    return FileResponse(CURRENT_DIR / "style.css", media_type="text/css")
+
+# JavaScript 파일 서빙
+@app.get("/script.js")
+async def serve_js():
+    return FileResponse(CURRENT_DIR / "script.js", media_type="application/javascript")
+
+# 루트 경로 ("/") 및 다른 모든 경로에 대해 index.html을 제공 (SPA 동작)
+# API 경로 (/api/...)는 이것보다 먼저 정의되어야 함
+@app.get("/{full_path:path}")
+async def serve_spa(request: Request, full_path: str):
+    # API 호출이 아닌 경우에만 index.html 반환
+    if full_path.startswith("api/"): # 이미 위에서 처리됨. 혹시 모를 경우 대비.
+        raise HTTPException(status_code=404, detail="API endpoint not found here.")
+    return FileResponse(CURRENT_DIR / "index.html", media_type="text/html")
+
+# 만약 위 방식이 복잡하다면, StaticFiles를 하나로 마운트하고 html=True 사용:
+# app.mount("/", StaticFiles(directory=CURRENT_DIR, html=True), name="static_root")
+# 이 경우, index.html, style.css, script.js 모두 CURRENT_DIR에서 직접 서빙됨.
+# /style.css 요청 시 ./style.css 파일 서빙.
+# / 요청 시 ./index.html 파일 서빙.
+# 이 방식이 더 간단하며, 위 개별 라우트는 필요 없을 수 있습니다.
+# 둘 중 하나를 선택하세요. 여기서는 더 명시적인 개별 라우트 방식을 남겨두었습니다.
+# StaticFiles(directory=".") 로 현재 디렉토리를 지정해도 됩니다.
+
+if __name__ == "__main__":
+    import uvicorn
+    # Hugging Face Spaces에서는 CMD 명령어로 uvicorn이 실행되므로 이 부분은 로컬 테스트용입니다.
+    # 포트 7860은 Hugging Face Spaces의 기본 포트입니다.
+    print(f"로컬 테스트 서버 시작: http://localhost:7860")
+    print(f"index.html은 다음 경로에 있어야 합니다: {CURRENT_DIR / 'index.html'}")
+    uvicorn.run(app, host="0.0.0.0", port=7860)