Update app.py

app.py

@@ -27,6 +27,11 @@ class Technology(BaseModel):
     limitations: str
     id: int
 
+class PriorArt(BaseModel):
+    """Represents the search of prior art using the technology combinations"""
+    content: str
+    uris: List
+
 # This schema defines the root structure of the JSON
 class TechnologyData(BaseModel):
     """Represents the top-level object containing a list of technologies."""
@@ -34,14 +39,23 @@ class TechnologyData(BaseModel):
 
 @app.post("/process", response_model=TechnologyData)
 async def process(data: InputData):
-    result = process_input(data, global_tech, global_tech_embeddings)
-    return {"technologies": result}
-
+    result, prior_art = process_input(data, global_tech, global_tech_embeddings, "problem")
+    return {"technologies": result, "prior_art": prior_art}
 
 @app.post("/process-constraints", response_model=TechnologyData)
 async def process_constraints(constraints: InputConstraints):
-    result = process_input_from_constraints(constraints.constraints, global_tech, global_tech_embeddings)
-    return {"technologies": result}
+    result, prior_art = process_input(constraints.constraints, global_tech, global_tech_embeddings, "constraints")
+    return {"technologies": result, "prior_art": prior_art}
+
+@app.post("/prior-art-constraints", response_model=PriorArt)
+async def prior_art_search(technologies: TechnologyData, constraints: InputConstraints, type: str):
+    prior_art = process_prior_art(technologies, constraints, type)
+    return prior_art
+
+@app.post("/prior-art-problems", response_model=PriorArt)
+async def prior_art_search(technologies: TechnologyData, problems: InputData, type: str):
+    prior_art = process_prior_art(technologies, problems, type)
+    return prior_art
 
 
 def make_json_serializable(data):