msg

main.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, HTTPException
-from services.sms_service import predict_label, compute_cosine_similarity, compute_embeddings
+from services.sms_service import label_prediction, compute_cosine_similarity, embeddings
 from schemas.input_schemas import CosineSimilarityInput, MessageInput, EmbeddingInput
 
 app = FastAPI()
@@ -7,7 +7,7 @@ app = FastAPI()
 # 🚀 1️⃣ Homepage Endpoint
 @app.get("/")
 async def home():
-    return {"message": "Welcome to SMS Classification API"}
+    return {"message": "Welcome to SMS Classification Dashboard"}
 
 # 🔢 2️⃣ Cosine Similarity Endpoint
 @app.post("/cosine_similarity")
@@ -18,17 +18,17 @@ async def get_cosine_similarity(input_data: CosineSimilarityInput):
         raise HTTPException(status_code=500, detail=f"Error computing similarity: {str(e)}")
 
 # 📩 3️⃣ SMS Classification Endpoint
-@app.post("/predict_label")
+@app.post("/label_prediction")
 async def classify_message(input_data: MessageInput):
     try:
-        return await predict_label(input_data.message)
+        return await label_prediction(input_data.message)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error predicting label: {str(e)}")
 
 # 📊 4️⃣ Text Embedding Endpoint
-@app.post("/compute_embeddings")
+@app.post("/embeddings")
 async def get_embeddings(input_data: EmbeddingInput):
     try:
-        return await compute_embeddings(input_data.message)
+        return await embeddings(input_data.message)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error computing embeddings: {str(e)}")