Update app.py

app.py

@@ -1,27 +1,29 @@
-import gradio as gr
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+import os
 from cad_model_generator import create_press_tool_model
-from model_validation import validate_model_with_ansys
 
-def generate_and_validate_model(length, width, height):
-    # Generate CAD model
-    stl_file = create_press_tool_model(length, width, height)
+# Initialize FastAPI app
+app = FastAPI()
 
-    # Validate the model with ANSYS
-    validation_results = validate_model_with_ansys(stl_file)
+# Input model for the API endpoint
+class ModelInput(BaseModel):
+    length: float
+    width: float
+    height: float
+
+# Endpoint to generate the CAD model
+@app.post("/generate-model/")
+async def generate_model(model_input: ModelInput):
+    try:
+        # Generate the CAD model using FreeCAD
+        model_path = create_press_tool_model(model_input.length, model_input.width, model_input.height)
+
+        # Return the path or URL to the generated STL file
+        return {"model_url": model_path}
     
-    return validation_results
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
-# Create Gradio interface
-iface = gr.Interface(
-    fn=generate_and_validate_model,
-    inputs=[
-        gr.Slider(minimum=10, maximum=500, label="Length (mm)"),
-        gr.Slider(minimum=10, maximum=500, label="Width (mm)"),
-        gr.Slider(minimum=5, maximum=100, label="Height (mm)"),
-    ],
-    outputs="text",
-    title="Press Tool CAD Model Generator and Validator",
-    description="Generate 3D press tool CAD models and validate them using ANSYS"
-)
+# To run the server with: uvicorn api:app --reload
 
-iface.launch()