2

app.py

@@ -1,9 +1,13 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
 import uvicorn
-from translator import translate
+from translator import translate  # Pastikan file translator.py ada
 
-app = FastAPI()
+app = FastAPI(
+    title="Chatbot & Translator API",
+    description="API untuk chatbot dan translator Indonesia-English",
+    version="1.0.0"
+)
 
 # ===== Model untuk request/response =====
 class ChatRequest(BaseModel):
@@ -14,22 +18,72 @@ class ChatRequest(BaseModel):
     temperature: float = 0.7
     top_p: float = 0.95
 
+class ChatResponse(BaseModel):
+    response: str
+    status: str = "success"
+
 class TranslateRequest(BaseModel):
     text: str
     direction: str  # "ID → EN" atau "EN → ID"
 
+class TranslateResponse(BaseModel):
+    translation: str
+    original_text: str
+    direction: str
+    status: str = "success"
+
+# ===== Root endpoint =====
+@app.get("/")
+def root():
+    return {
+        "message": "Chatbot & Translator API", 
+        "endpoints": ["/chat", "/translate"],
+        "docs": "/docs"
+    }
+
 # ===== Endpoint Chatbot =====
-@app.post("/chat")
+@app.post("/chat", response_model=ChatResponse)
 def chat_endpoint(req: ChatRequest):
-    # ⚡ Di sini kamu bisa sambungkan dengan InferenceClient seperti di kode awal
-    return {"response": f"Simulasi jawaban untuk: {req.message}"}
+    try:
+        # TODO: Implementasi actual chatbot logic di sini
+        # Contoh dengan InferenceClient atau model lainnya
+        response_text = f"Simulasi jawaban untuk: {req.message}"
+        
+        return ChatResponse(
+            response=response_text,
+            status="success"
+        )
+    except Exception as e:
+        return ChatResponse(
+            response=f"Error: {str(e)}",
+            status="error"
+        )
 
 # ===== Endpoint Translator =====
-@app.post("/translate")
+@app.post("/translate", response_model=TranslateResponse)
 def translate_endpoint(req: TranslateRequest):
-    result = translate(req.text, req.direction)
-    return {"translation": result}
+    try:
+        result = translate(req.text, req.direction)
+        
+        return TranslateResponse(
+            translation=result,
+            original_text=req.text,
+            direction=req.direction,
+            status="success"
+        )
+    except Exception as e:
+        return TranslateResponse(
+            translation="Translation failed",
+            original_text=req.text,
+            direction=req.direction,
+            status="error"
+        )
+
+# ===== Health check endpoint =====
+@app.get("/health")
+def health_check():
+    return {"status": "healthy", "message": "API is running"}
 
 # ====== Run local (Space akan otomatis pakai uvicorn) ======
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)