Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import requests
-import os
 
 # Hugging Face Model API URL
 HF_MODEL_URL = "https://api-inference.huggingface.co/models/your_model_here"
@@ -8,7 +7,6 @@ HF_API_KEY = "YOUR_HUGGINGFACE_API_KEY"
 
 # Function to get the concrete mix design from the Hugging Face model API
 def get_concrete_mix_design(strength, slump, w_c_ratio, coarse_aggregate, fine_aggregate, mix_code):
-    # Prepare the payload based on selected mix code
     payload = {
         "inputs": {
             "strength": strength,
@@ -16,28 +14,28 @@ def get_concrete_mix_design(strength, slump, w_c_ratio, coarse_aggregate, fine_a
             "w_c_ratio": w_c_ratio,
             "coarse_aggregate": coarse_aggregate,
             "fine_aggregate": fine_aggregate,
-            "mix_code": mix_code  # Add the selected mix design code here
+            "mix_code": mix_code
         }
     }
-    
+
     headers = {"Authorization": f"Bearer {HF_API_KEY}"}
-    response = requests.post(HF_MODEL_URL, headers=headers, json=payload)
+    try:
+        response = requests.post(HF_MODEL_URL, headers=headers, json=payload)
+        response.raise_for_status()  # Raise an HTTPError for bad responses (4xx, 5xx)
 
-    if response.status_code == 200:
-        result = response.json()
-        mix_design = result.get('mix_design', 'No design available.')
-        return mix_design
-    else:
-        return "Error in getting mix design. Please try again."
+        if response.status_code == 200:
+            result = response.json()
+            mix_design = result.get('mix_design', 'No design available.')
+            return format_mix_design(mix_design)
+        else:
+            return f"Error: Received status code {response.status_code}. Response: {response.text}"
+    except requests.exceptions.RequestException as e:
+        return f"Request failed: {e}"
 
-# Function to handle the user interface logic
-def concrete_mix_ui(strength, slump, w_c_ratio, coarse_aggregate, fine_aggregate, mix_code):
-    # Get the mix design based on the user's input and selected mix code
-    mix_design = get_concrete_mix_design(strength, slump, w_c_ratio, coarse_aggregate, fine_aggregate, mix_code)
-    
-    # Format the output
+# Function to format the mix design output
+def format_mix_design(mix_design):
     if isinstance(mix_design, dict):  # Check if we received valid data
-        return f"Recommended Concrete Mix Design (Using {mix_code}):\n\n" \
+        return f"Recommended Concrete Mix Design (Using {mix_design.get('mix_code', 'Unknown')}):\n\n" \
                f"- Cement: {mix_design.get('cement', 'N/A')} kg/m³\n" \
                f"- Water: {mix_design.get('water', 'N/A')} liters/m³\n" \
                f"- Fine Aggregate: {mix_design.get('fine_aggregate', 'N/A')} kg/m³\n" \
@@ -46,7 +44,13 @@ def concrete_mix_ui(strength, slump, w_c_ratio, coarse_aggregate, fine_aggregate
                f"- Water-Cement Ratio: {mix_design.get('w_c_ratio', 'N/A')}\n" \
                f"- Strength: {mix_design.get('strength', 'N/A')} MPa"
     else:
-        return mix_design  # If the design is not available, return the error message
+        return "No valid design returned."
+
+# Function to handle the user interface logic
+def concrete_mix_ui(strength, slump, w_c_ratio, coarse_aggregate, fine_aggregate, mix_code):
+    # Get the mix design based on the user's input and selected mix code
+    mix_design = get_concrete_mix_design(strength, slump, w_c_ratio, coarse_aggregate, fine_aggregate, mix_code)
+    return mix_design
 
 # User Interface: Define the input components
 inputs = [
@@ -72,5 +76,6 @@ interface = gr.Interface(
     theme="compact",
 )
 
-# Launch the app
+# Launch the app with public link
 interface.launch(share=True)
+