Create app.py

app.py

@@ -0,0 +1,97 @@
+import gradio as gr
+
+# Assume your existing functions are here (collect_input, check_pump_suitability, etc.)
+def collect_input():
+    """
+    Collects necessary input data from the user.
+    """
+    print("Please enter the following data about the well:")
+    depth = float(input("Well Depth (meters): "))
+    viscosity = float(input("Fluid Viscosity (cP): "))
+    flow_rate = float(input("Flow Rate (m³/day): "))
+    specific_gravity = float(input("Specific Gravity of Fluid: "))
+    temperature = float(input("Temperature (Celsius): "))
+    solids_content = float(input("Percentage of Solids (%): "))
+    corrosivity = input("Is the fluid corrosive? (yes/no): ").lower() == 'yes'
+
+    return {
+        "depth": depth,
+        "viscosity": viscosity,
+        "flow_rate": flow_rate,
+        "specific_gravity": specific_gravity,
+        "temperature": temperature,
+        "solids_content": solids_content,
+        "corrosivity": corrosivity
+    }
+
+def check_pump_suitability(well_data):
+    """
+    Determines the suitability of a progressive cavity pump based on well data.
+    """
+    # Example criteria (these values can be adjusted based on real-world data and expert advice)
+    max_viscosity = 10000  # cP
+    max_solids_content = 10  # Percentage
+
+    if well_data["viscosity"] > max_viscosity:
+        return False, "Viscosity too high for progressive cavity pump."
+    if well_data["solids_content"] > max_solids_content:
+        return False, "Solids content too high for progressive cavity pump."
+    if well_data["corrosivity"]:
+        return False, "Corrosive fluid not suitable for standard progressive cavity pump."
+
+    # Further checks can be added here...
+
+    return True, "Well is suitable for a progressive cavity pump."
+
+def display_results(suitability, message):
+    """
+    Displays the results of the suitability check.
+    """
+    if suitability:
+        print("Result: Suitable.")
+    else:
+        print("Result: Not Suitable.")
+    print("Reason: ", message)
+
+def main():
+    well_data = collect_input()
+    suitability, message = check_pump_suitability(well_data)
+    display_results(suitability, message)
+
+if __name__ == "__main__":
+    main()
+
+def assess_pump_suitability(depth, viscosity, flow_rate, specific_gravity, temperature, solids_content, corrosivity):
+    well_data = {
+        "depth": depth,
+        "viscosity": viscosity,
+        "flow_rate": flow_rate,
+        "specific_gravity": specific_gravity,
+        "temperature": temperature,
+        "solids_content": solids_content,
+        "corrosivity": corrosivity == 'Yes'
+    }
+    suitability, message = check_pump_suitability(well_data)
+    return "Suitable" if suitability else "Not Suitable", message
+
+iface = gr.Interface(
+    fn=assess_pump_suitability,
+    inputs=[
+        gr.Number(label="Well Depth (meters)"),
+        gr.Number(label="Fluid Viscosity (cP)"),
+        gr.Number(label="Flow Rate (m³/day)"),
+        gr.Number(label="Specific Gravity of Fluid"),
+        gr.Number(label="Temperature (Celsius)"),
+        gr.Number(label="Percentage of Solids (%)"),
+        gr.Radio(["Yes", "No"], label="Is the fluid corrosive?")
+    ],
+    outputs=[
+        gr.Textbox(label="Suitability"),
+        gr.Textbox(label="Reason")
+    ],
+    title="Progressive Cavity Pump Suitability Assessment",
+    description="Enter well data to assess if it's suitable for a progressive cavity pump."
+)
+
+if __name__ == "__main__":
+    iface.launch(share=True)