Update app.py

app.py

@@ -17,16 +17,28 @@ class InputData(BaseModel):
 class InputConstraints(BaseModel):
     constraints: Dict[str, str]
     
-class OutputData(BaseModel):
-    technologies: list
-
-@app.post("/process", response_model=OutputData)
+# This schema defines the structure for a single technology object
+class Technology(BaseModel):
+    """Represents a single technology entry with its details."""
+    title: str
+    purpose: str
+    key_components: str
+    advantages: str
+    limitations: str
+    id: int
+
+# This schema defines the root structure of the JSON
+class TechnologyData(BaseModel):
+    """Represents the top-level object containing a list of technologies."""
+    technologies: List[Technology]
+
+@app.post("/process", response_model=TechnologyData)
 async def process(data: InputData):
     result = process_input(data, global_tech, global_tech_embeddings)
     return {"technologies": result}
 
 
-@app.post("/process-constraints", response_model=OutputData)
+@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}