Update app.py

app.py

@@ -176,13 +176,38 @@ def predictmic(sequence):
 
     return mic_results
 
-# Gradio interface
+
+
+
+
+def full_prediction(sequence):
+    # AMP prediction
+    features = extract_features(sequence)
+    if isinstance(features, str) and features.startswith("Error:"):
+        return "Error", 0.0, {}
+
+    prediction = model.predict(features)[0]
+    probabilities = model.predict_proba(features)[0]
+
+    amp_result = "Antimicrobial Peptide (AMP)" if prediction == 0 else "Non-AMP"
+    confidence = round(probabilities[0 if prediction == 0 else 1] * 100, 2)
+
+    # MIC prediction
+    mic_values = predictmic(sequence)
+
+    return amp_result, f"{confidence}%", mic_values
+
+import gradio as gr
 iface = gr.Interface(
-    fn=predict,
+    fn=full_prediction,
     inputs=gr.Textbox(label="Enter Protein Sequence"),
-    outputs=gr.Label(label="Prediction"),
-    title="AMP Classifier",
-    description="Enter an amino acid sequence (e.g., FLPVLAGGL) to predict AMP."
+    outputs=[
+        gr.Label(label="AMP Classification"),
+        gr.Label(label="Confidence"),
+        gr.JSON(label="Predicted MIC (µg/mL) for Each Bacterium")
+    ],
+    title="AMP & MIC Predictor",
+    description="Enter an amino acid sequence (e.g., FLPVLAGGL) to predict AMP class and MIC values across bacteria."
 )
 
-iface.launch(share=True)
+iface.launch(share=True)