Spaces:
Sleeping
Sleeping
File size: 7,007 Bytes
781e8cf 990e743 781e8cf 59cc5ec 781e8cf 59cc5ec 781e8cf 990e743 59cc5ec 781e8cf 59cc5ec 781e8cf 59cc5ec 781e8cf 59cc5ec 781e8cf 59cc5ec 781e8cf 59cc5ec 781e8cf 59cc5ec 781e8cf 59cc5ec 781e8cf 59cc5ec 781e8cf 59cc5ec | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 | from fastapi import FastAPI
from fastapi.responses import HTMLResponse
from pydantic import BaseModel
from typing import List, Any
import joblib
import pandas as pd
import numpy as np
app = FastAPI(title="Getaround Pricing API")
# ── Load model ────────────────────────────────────────────────────────────
try:
model = joblib.load("pricing_model.joblib")
except:
model = None
# ── Input schema ──────────────────────────────────────────────────────────
class PredictInput(BaseModel):
input: List[List[Any]]
COLUMNS = [
"model_key", "mileage", "engine_power", "fuel", "paint_color",
"car_type", "private_parking_available", "has_gps",
"has_air_conditioning", "automatic_car", "has_getaround_connect",
"has_speed_regulator", "winter_tires"
]
# ── /predict ──────────────────────────────────────────────────────────────
@app.post("/predict")
def predict(data: PredictInput):
df = pd.DataFrame(data.input, columns=COLUMNS)
for col in ["mileage", "engine_power"]:
df[col] = df[col].astype(float)
for col in ["private_parking_available", "has_gps", "has_air_conditioning",
"automatic_car", "has_getaround_connect", "has_speed_regulator", "winter_tires"]:
df[col] = df[col].astype(bool)
predictions = model.predict(df).tolist()
return {"prediction": predictions}
# ── /documentation — page HTML custom ────────────────────────────────────
@app.get("/documentation", response_class=HTMLResponse)
def documentation():
return """
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Getaround API Documentation</title>
<style>
@import url('https://fonts.googleapis.com/css2?family=DM+Sans:wght@400;600;700;800&display=swap');
* { margin: 0; padding: 0; box-sizing: border-box; }
body { font-family: 'DM Sans', sans-serif; background: #FAFAFA; color: #1A1A1A; }
header { background: #fff; border-bottom: 2px solid #E8DFF0; padding: 24px 48px; display: flex; align-items: center; gap: 12px; }
header h1 { font-size: 24px; font-weight: 800; }
header span { color: #7B2D8B; }
header .badge { background: #7B2D8B; color: white; font-size: 11px; padding: 4px 10px; border-radius: 20px; font-weight: 600; letter-spacing: 1px; }
main { max-width: 860px; margin: 48px auto; padding: 0 24px; }
h2 { font-size: 20px; font-weight: 800; margin-bottom: 8px; color: #1A1A1A; }
.subtitle { color: #888; font-size: 14px; margin-bottom: 40px; }
.endpoint { background: white; border: 1px solid #E8DFF0; border-radius: 12px; padding: 28px; margin-bottom: 24px; }
.endpoint-header { display: flex; align-items: center; gap: 12px; margin-bottom: 16px; }
.method { background: #7B2D8B; color: white; font-size: 12px; font-weight: 700; padding: 4px 12px; border-radius: 6px; letter-spacing: 1px; }
.method.get { background: #2D8B6B; }
.path { font-size: 18px; font-weight: 700; font-family: monospace; }
.desc { color: #555; font-size: 14px; margin-bottom: 16px; }
.section-label { font-size: 11px; font-weight: 700; text-transform: uppercase; letter-spacing: 1px; color: #7B2D8B; margin-bottom: 8px; margin-top: 16px; }
pre { background: #F5F0F7; border-radius: 8px; padding: 16px; font-size: 13px; overflow-x: auto; line-height: 1.6; }
table { width: 100%; border-collapse: collapse; font-size: 13px; margin-top: 8px; }
th { background: #F5F0F7; padding: 10px 14px; text-align: left; font-weight: 700; color: #7B2D8B; }
td { padding: 10px 14px; border-bottom: 1px solid #E8DFF0; }
tr:last-child td { border-bottom: none; }
.note { background: #F5F0F7; border-left: 4px solid #7B2D8B; border-radius: 0 8px 8px 0; padding: 12px 16px; font-size: 13px; color: #555; margin-top: 16px; }
</style>
</head>
<body>
<header>
<h1>get<span>around</span></h1>
<div class="badge">API DOCS</div>
</header>
<main>
<h2>Getaround Pricing API</h2>
<p class="subtitle">API de prédiction de prix pour l'optimisation tarifaire des véhicules Getaround.</p>
<div class="endpoint">
<div class="endpoint-header">
<span class="method">POST</span>
<span class="path">/predict</span>
</div>
<p class="desc">Prédit le prix optimal par jour pour un ou plusieurs véhicules.</p>
<div class="section-label">Ordre des valeurs par véhicule</div>
<table>
<tr><th>#</th><th>Champ</th><th>Type</th><th>Exemple</th></tr>
<tr><td>0</td><td>model_key</td><td>string</td><td>"Renault"</td></tr>
<tr><td>1</td><td>mileage</td><td>float</td><td>80000</td></tr>
<tr><td>2</td><td>engine_power</td><td>float</td><td>120</td></tr>
<tr><td>3</td><td>fuel</td><td>string</td><td>"diesel"</td></tr>
<tr><td>4</td><td>paint_color</td><td>string</td><td>"black"</td></tr>
<tr><td>5</td><td>car_type</td><td>string</td><td>"sedan"</td></tr>
<tr><td>6</td><td>private_parking_available</td><td>bool</td><td>true</td></tr>
<tr><td>7</td><td>has_gps</td><td>bool</td><td>true</td></tr>
<tr><td>8</td><td>has_air_conditioning</td><td>bool</td><td>true</td></tr>
<tr><td>9</td><td>automatic_car</td><td>bool</td><td>false</td></tr>
<tr><td>10</td><td>has_getaround_connect</td><td>bool</td><td>true</td></tr>
<tr><td>11</td><td>has_speed_regulator</td><td>bool</td><td>true</td></tr>
<tr><td>12</td><td>winter_tires</td><td>bool</td><td>false</td></tr>
</table>
<div class="section-label">Exemple de requête</div>
<pre>curl -X POST https://nana12a-getaround-api.hf.space/predict \
-H "Content-Type: application/json" \
-d '{"input": [["Renault", 80000, 120, "diesel", "black", "sedan", true, true, true, false, true, true, false]]}'</pre>
<div class="section-label">Exemple de réponse</div>
<pre>{"prediction": [143.43]}</pre>
<div class="note">💡 Swagger interactif disponible sur <a href="/docs">/docs</a></div>
</div>
<div class="endpoint">
<div class="endpoint-header">
<span class="method get">GET</span>
<span class="path">/health</span>
</div>
<p class="desc">Vérifie que l'API est en ligne et que le modèle est bien chargé.</p>
<div class="section-label">Exemple de réponse</div>
<pre>{"status": "ok", "model": "loaded"}</pre>
</div>
</main>
</body>
</html>
"""
# ── /health ───────────────────────────────────────────────────────────────
@app.get("/health")
def health():
return {"status": "ok", "model": "loaded" if model else "unavailable"}
|