changed

main.py

@@ -1,7 +1,6 @@
-# app/main.py
 from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
 from services.sms_service import predict_label, compute_cosine_similarity, compute_embeddings
+from schemas.input_schemas import CosineSimilarityInput, MessageInput, EmbeddingInput
 
 app = FastAPI()
 
@@ -12,7 +11,7 @@ async def home():
 
 # 🔢 2️⃣ Cosine Similarity Endpoint
 @app.post("/cosine_similarity")
-async def get_cosine_similarity(input_data: BaseModel):
+async def get_cosine_similarity(input_data: CosineSimilarityInput):
     try:
         return await compute_cosine_similarity(input_data.text1, input_data.text2)
     except Exception as e:
@@ -20,7 +19,7 @@ async def get_cosine_similarity(input_data: BaseModel):
 
 # 📩 3️⃣ SMS Classification Endpoint
 @app.post("/predict_label")
-async def classify_message(input_data: BaseModel):
+async def classify_message(input_data: MessageInput):
     try:
         return await predict_label(input_data.message)
     except Exception as e:
@@ -28,7 +27,7 @@ async def classify_message(input_data: BaseModel):
 
 # 📊 4️⃣ Text Embedding Endpoint
 @app.post("/compute_embeddings")
-async def get_embeddings(input_data: BaseModel):
+async def get_embeddings(input_data: EmbeddingInput):
     try:
         return await compute_embeddings(input_data.message)
     except Exception as e: