Update app.py

app.py

@@ -1,140 +1,92 @@
 import gradio as gr
-import subprocess
 import os
+import subprocess
 
-# APDL Script for ANSYS
-apdl_script = """/CLEAR            ! Clear any existing database
-/PREP7            ! Enter preprocessor
-/CWD, '{cwd}'     ! Set the working directory
-! Define material properties
-MP, EX, 1, 210E3  ! Young's Modulus (e.g., Steel)
-MP, PRXY, 1, 0.3  ! Poisson's Ratio
-! Define geometry (e.g., a rectangular plate)
-RECTNG, 0, 100, 0, 50  ! Rectangle from (0,0) to (100,50)
-! Define element type and mesh
-ET, 1, PLANE183   ! 2D structural element
-ESIZE, 5          ! Set element size
-AMESH, ALL        ! Mesh the area
-! Apply boundary conditions
-D, NODE(0, 0), ALL, 0     ! Fix the bottom-left corner
-D, NODE(100, 0), UY, 0    ! Restrict vertical movement at bottom-right
-! Apply loads
-F, NODE(50, 50), FX, 1000 ! Apply a horizontal load at the top-middle node
-! Solve
-FINISH           ! Exit preprocessor
-/SOLU            ! Enter solution processor
-ANTYPE, 0        ! Static analysis
-SOLVE            ! Solve the system
-FINISH           ! Exit solution processor
-/POST1           ! Enter post-processor
-/SHOW, JPEG          ! Set output format to JPEG
-/DEVICE, HARD, JPEG  ! Configure device for image output
-PLNSOL, U, SUM       ! Plot nodal displacement (2D view)
-FILE, '2D_View', 'JPEG'  ! Save 2D view as an image
-/SHOW, CLOSE         ! Close the graphical window
-/VIEW, 1, 1, 1, 1    ! Set 3D view (isometric)
-/SHOW, JPEG          ! Set output format to JPEG
-/DEVICE, HARD, JPEG  ! Configure device for image output
-PLNSOL, S, EQV       ! Plot equivalent stress (3D view)
-FILE, '3D_View', 'JPEG'  ! Save 3D view as an image
-/SHOW, CLOSE         ! Close the graphical window
-FINISH
+# Default script template for FreeCAD
+freecad_script = """import FreeCAD
+import Part
+
+# Open STEP file
+doc = FreeCAD.newDocument("MyDocument")
+shape = Part.Shape()
+shape.read("{step_file}")
+part = doc.addObject("Part::Feature", "ImportedPart")
+part.Shape = shape
+
+# Save as a new STEP file
+doc.recompute()
+doc.saveAs("{output_file}")
+print(f"STEP file processed and saved to {output_file}")
 """
 
 def upload_step_file(file):
     """Handle STEP file upload."""
-    return f"STEP file '{file.name}' uploaded successfully."
-
-def modify_apdl_script(modified_script):
-    """Modify and save the APDL script."""
-    global apdl_script
-    apdl_script = modified_script
-    return "APDL script updated successfully."
-
-def execute_script():
-    """Execute the APDL script using ANSYS."""
-    global apdl_script
-
-    # Set the working directory
-    cwd = os.getcwd()
-    apdl_script_with_cwd = apdl_script.format(cwd=cwd)
-
-    # Save the APDL script to a file
-    script_file = "script.dat"
+    file_path = os.path.join(os.getcwd(), file.name)
+    with open(file_path, "wb") as f:
+        f.write(file.read())
+    return f"STEP file '{file.name}' uploaded successfully.", file_path
+
+def modify_script(script_text):
+    """Modify and save the FreeCAD script."""
+    global freecad_script
+    freecad_script = script_text
+    script_file = "modified_script.py"
     with open(script_file, "w") as f:
-        f.write(apdl_script_with_cwd)
-
-    # Define ANSYS execution command
-    ansys_command = [
-        "C:\\Program Files\\ANSYS Inc\\v211\\ansys\\bin\\winx64\\ansys.exe",  # Replace with your actual path
-        "-b",
-        "-i", script_file,
-        "-o", "output.log"
+        f.write(freecad_script)
+    return "FreeCAD script updated successfully.", script_file
+
+def execute_freecad_script(step_file):
+    """Execute the FreeCAD script."""
+    global freecad_script
+    script_path = "temp_script.py"
+    output_file = "output_processed.step"
+
+    # Fill the script with file paths
+    script_with_paths = freecad_script.format(
+        step_file=step_file, output_file=output_file
+    )
+    with open(script_path, "w") as f:
+        f.write(script_with_paths)
+
+    # Command to run FreeCAD
+    freecad_command = [
+        "C:\\Program Files\\FreeCAD 1.0\\bin\\FreeCADCmd.exe",
+        script_path,
     ]
 
     try:
-        # Run the ANSYS command
-        subprocess.run(ansys_command, check=True)
-
-        # Check for image outputs
-        two_d_image = os.path.join(cwd, "2D_View.jpg")
-        three_d_image = os.path.join(cwd, "3D_View.jpg")
-
-        if os.path.exists(two_d_image) and os.path.exists(three_d_image):
-            return "APDL script executed successfully. 2D and 3D views generated."
+        subprocess.run(freecad_command, check=True)
+        if os.path.exists(output_file):
+            return f"Script executed successfully. Processed STEP file saved at {output_file}.", output_file
         else:
-            with open("output.log", "r") as log_file:
-                log_contents = log_file.read()
-            return f"APDL script executed, but images not found. Check log:\n{log_contents}"
-
+            return "Script executed, but no output file found.", None
     except subprocess.CalledProcessError as e:
-        return f"Error executing APDL script: {str(e)}. Check output.log for details."
+        return f"Error executing FreeCAD script: {e}", None
     except FileNotFoundError:
-        return "ANSYS executable not found. Verify the path in ansys_command."
-
-def get_generated_images():
-    """Return the generated 2D and 3D view images if they exist."""
-    two_d_image = os.path.join(os.getcwd(), "2D_View.jpg")
-    three_d_image = os.path.join(os.getcwd(), "3D_View.jpg")
-    if os.path.exists(two_d_image) and os.path.exists(three_d_image):
-        return two_d_image, three_d_image
-    else:
-        return None, None
-
-def read_output_log():
-    """Read and return the contents of output.log."""
-    if os.path.exists("output.log"):
-        with open("output.log", "r") as f:
-            return f.read()
-    else:
-        return "No output log found."
+        return "FreeCAD executable not found. Verify the path in freecad_command.", None
 
 # Gradio interface
 with gr.Blocks() as app:
-    gr.Markdown("# ANSYS APDL Script for 2D and 3D Views")
+    gr.Markdown("# FreeCAD Script Processor")
 
     with gr.Row():
         step_file = gr.File(label="Upload STEP File")
-        step_output = gr.Text(label="Upload Status")
-        step_file.upload(upload_step_file, inputs=step_file, outputs=step_output)
+        upload_status = gr.Text(label="Upload Status")
+        step_path = gr.Text(label="Uploaded File Path (Internal Use)", visible=False)
+        step_file.upload(upload_step_file, inputs=step_file, outputs=[upload_status, step_path])
 
     with gr.Row():
-        apdl_view = gr.Textbox(apdl_script, lines=20, label="APDL Script")
-        modify_button = gr.Button("Save Changes")
-        modify_button.click(modify_apdl_script, inputs=apdl_view, outputs=step_output)
+        script_view = gr.Textbox(freecad_script, lines=20, label="FreeCAD Script")
+        save_button = gr.Button("Save Script")
+        script_status = gr.Text(label="Script Status")
+        modified_script = gr.File(label="Download Modified Script")
+        save_button.click(modify_script, inputs=script_view, outputs=[script_status, modified_script])
 
     with gr.Row():
-        execute_button = gr.Button("Run Script")
-        execution_output = gr.Text(label="Execution Status")
-        log_output = gr.Textbox(label="Output Log", lines=10, interactive=False)
-        execute_button.click(execute_script, outputs=[execution_output])
-        execute_button.click(read_output_log, outputs=log_output)
-
-    with gr.Row():
-        gr.Markdown("## Generated Images")
-        view_2d = gr.Image(label="2D View", interactive=False)
-        view_3d = gr.Image(label="3D View", interactive=False)
-        refresh_button = gr.Button("Refresh Images")
-        refresh_button.click(get_generated_images, outputs=[view_2d, view_3d])
+        execute_button = gr.Button("Run FreeCAD Script")
+        execution_status = gr.Text(label="Execution Status")
+        output_file = gr.File(label="Download Processed STEP File")
+        execute_button.click(execute_freecad_script, inputs=step_path, outputs=[execution_status, output_file])
 
 app.launch()
+