Update app.py

app.py

@@ -1,20 +1,17 @@
 import gradio as gr
 import requests
 
-# Hugging Face Model API URL
-HF_MODEL_URL = "https://api-inference.huggingface.co/models/your_model_here"
-HF_API_KEY = "YOUR_HUGGINGFACE_API_KEY"
+# Using a general-purpose model hosted on Hugging Face
+HF_MODEL_URL = "https://api-inference.huggingface.co/models/distilbert-base-uncased"
+HF_API_KEY = "YOUR_HUGGINGFACE_API_KEY"  # Make sure your API key is valid
 
-# Function to get the concrete mix design from the Hugging Face model API
+# Function to get a response from the Hugging Face model API
 def get_concrete_mix_design(strength, slump, w_c_ratio, coarse_aggregate, fine_aggregate, mix_code):
     payload = {
         "inputs": {
-            "strength": strength,
-            "slump": slump,
-            "w_c_ratio": w_c_ratio,
-            "coarse_aggregate": coarse_aggregate,
-            "fine_aggregate": fine_aggregate,
-            "mix_code": mix_code
+            "text": f"Strength: {strength} MPa, Slump: {slump} mm, Water-Cement Ratio: {w_c_ratio}, "
+                    f"Coarse Aggregate: {coarse_aggregate} kg/m³, Fine Aggregate: {fine_aggregate} kg/m³, "
+                    f"Mix Code: {mix_code}"
         }
     }
 
@@ -25,27 +22,13 @@ def get_concrete_mix_design(strength, slump, w_c_ratio, coarse_aggregate, fine_a
 
         if response.status_code == 200:
             result = response.json()
-            mix_design = result.get('mix_design', 'No design available.')
-            return format_mix_design(mix_design)
+            generated_text = result.get('generated_text', 'No response from model.')
+            return generated_text
         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 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_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" \
-               f"- Coarse Aggregate: {mix_design.get('coarse_aggregate', 'N/A')} kg/m³\n" \
-               f"- Admixture: {mix_design.get('admixture', 'N/A')} liters/m³\n" \
-               f"- Water-Cement Ratio: {mix_design.get('w_c_ratio', 'N/A')}\n" \
-               f"- Strength: {mix_design.get('strength', 'N/A')} MPa"
-    else:
-        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
@@ -79,3 +62,4 @@ interface = gr.Interface(
 # Launch the app with public link
 interface.launch(share=True)
 
+