stable2

app.py

@@ -1,3 +1,4 @@
+#import library
 import gradio as gr
 from roboflow import Roboflow
 import numpy as np
@@ -51,8 +52,7 @@ def get_inspection_recommendation(total_area):
         return "Immediate inspection required."
 
 # Define a Gradio interface to input a URL, run inference, and calculate corrosion areas
-# Define a Gradio interface to input a URL, run inference, and calculate corrosion areas
-def url_infer_and_calculate(url, location=None, unit="pixels", conversion_factor=1, corrosion_type=None, inspection_standards=None, ndt_methods=None, recommendation=None, files=None):
+def url_infer_and_calculate(url, location, unit="pixels", conversion_factor=1):
     try:
         # Run inference using the Roboflow script
         rf_project = rf.workspace().project("corrosion-instance-segmentation-sfcpc")
@@ -75,23 +75,11 @@ def url_infer_and_calculate(url, location=None, unit="pixels", conversion_factor
         # Write DataFrame to local CSV file with index included immediately after creating it.
         df.to_csv('Corrosion_Report.csv', index=False)
 
-        # Add the new inputs to the result
-        result = {
-            "corrosion_type": corrosion_type,
-            "inspection_standards": inspection_standards,
-            "ndt_methods": ndt_methods,
-            "manual_recommendation": recommendation,
-            "files": [f.name for f in files] if files else []
-        }
-
-        # Return the path of the CSV file along with other outputs.
-        return img, corrosion_areas, prediction_json, os.path.abspath('Corrosion_Report.csv'), result
+        return img, corrosion_areas, prediction_json
 
     except Exception as e:
         return {"error": str(e)}
 
-
-
 # Create a Gradio interface for URL input, inference, and corrosion area calculation
 iface = gr.Interface(
     fn=url_infer_and_calculate,
@@ -99,17 +87,22 @@ iface = gr.Interface(
       gr.inputs.Textbox(label="Enter the URL of an image"),
       gr.inputs.Textbox(label="Enter the Location"),
       gr.inputs.Dropdown(choices=["pixels", "cm??"], label="Area Unit"),
-      gr.inputs.Textbox(label="Conversion Factor"),
-      gr.inputs.Textbox(label="Enter Corrosion Type"),
-      gr.inputs.CheckboxGroup(choices=[f"Standard {i+1}" for i in range(10)], label="Inspection Standards"),
-      gr.inputs.CheckboxGroup(choices=["UT thickness", "UT scan", "Phased Array UT", "Short range UT", "Long range UT", "MT", "PT", "Acfm", "Pulse eddy current", "magnetic flux leakage"], label="NDT Inspection Methods"),
-      gr.inputs.Textbox(label="Enter Manual Recommendation"),
-      *[gr.inputs.File(label=f"Upload Supporting Data {i+1}") for i in range(10)]
+      gr.inputs.Textbox(label="Conversion Factor")
     ],
-    outputs=[gr.outputs.Image(type="pil"), "json", "json", gr.outputs.Textbox(label='CSV File Path'), "json"],
+    outputs=[gr.outputs.Image(type="pil"), "json", "json"],
     title="Tim CCG",
     description="Enter the URL of an image to perform rust detection and calculate corrosion areas.",
 )
 
 # Launch the Gradio interface
 iface.launch(debug=False, share=False)
+
+
+
+
+
+
+
+
+
+