heymenn commited on
Commit
1eebfeb
·
verified ·
1 Parent(s): e602ae8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -50,24 +50,25 @@ class TechnologyData(BaseModel):
50
 
51
  @app.post("/process", response_model=TechnologyData)
52
  async def process(data: InputProblem):
53
- result, prior_art = process_input(data, global_tech, global_tech_embeddings, "problem")
54
- return {"technologies": result, "prior_art": prior_art}
55
 
56
  @app.post("/process-constraints", response_model=TechnologyData)
57
  async def process_constraints(constraints: InputConstraints):
58
- result, prior_art = process_input(constraints.constraints, global_tech, global_tech_embeddings, "constraints")
59
- return {"technologies": result, "prior_art": prior_art}
60
 
61
  @app.post("/prior-art-constraints", response_model=OutputPriorArt)
62
  async def prior_art_search(data: InputPriorArtConstraints):
63
- prior_art = process_prior_art(InputPriorArtConstraints.technologies, InputPriorArtConstraints.constriants, InputPriorArtConstraints.data_type)
64
  return prior_art
65
 
66
  @app.post("/prior-art-problems", response_model=OutputPriorArt)
67
  async def prior_art_search(data: InputPriorArtProblems):
68
- prior_art = process_prior_art(InputPriorArtProblems.technologies, InputPriorArtProblems.problems, InputPriorArtProblems.data_type)
69
  return prior_art
70
 
 
71
  def make_json_serializable(data):
72
  if isinstance(data, dict):
73
  return {k: make_json_serializable(v) for k, v in data.items()}
 
50
 
51
  @app.post("/process", response_model=TechnologyData)
52
  async def process(data: InputProblem):
53
+ result= process_input(data, global_tech, global_tech_embeddings, "problem")
54
+ return {"technologies": result}
55
 
56
  @app.post("/process-constraints", response_model=TechnologyData)
57
  async def process_constraints(constraints: InputConstraints):
58
+ result= process_input(constraints.constraints, global_tech, global_tech_embeddings, "constraints")
59
+ return {"technologies": result}
60
 
61
  @app.post("/prior-art-constraints", response_model=OutputPriorArt)
62
  async def prior_art_search(data: InputPriorArtConstraints):
63
+ prior_art = process_prior_art(data.technologies, data.constriants, data.data_type)
64
  return prior_art
65
 
66
  @app.post("/prior-art-problems", response_model=OutputPriorArt)
67
  async def prior_art_search(data: InputPriorArtProblems):
68
+ prior_art = process_prior_art(data.technologies, data.problems, data.data_type)
69
  return prior_art
70
 
71
+
72
  def make_json_serializable(data):
73
  if isinstance(data, dict):
74
  return {k: make_json_serializable(v) for k, v in data.items()}