Update app.py

app.py

@@ -19,10 +19,24 @@ async def root():
     return {"message": "Embedding API is running."}
 
 # Embedding endpoint
-@app.post("/embed")
-async def embed_text(request: EmbeddingRequest):
-    embeddings = model.encode(request.texts).tolist()
-    return {"embeddings": embeddings}
+@app.post("/get_sim_text")
+def get_sim_text_endpoint(request: TextPairRequest):
+    try:
+        # Embed the claim text
+        claim_embedding = similarity_model.model.encode(
+            request.claim,
+            convert_to_tensor=True,
+            show_progress_bar=False
+        )
+        # Call get_sim_text with defaults (min_threshold=0.4, chunk_size=1500)
+        results = similarity_model.get_sim_text(
+            request.text,
+            claim_embedding
+        )
+        return {"result": results}
+
+    except Exception as e:
+        return {"error": str(e)}
 
 # Optional for local testing
 if __name__ == "__main__":