Update app.py

app.py

@@ -109,18 +109,26 @@ def predictmic(sequence):
 # Combined Output as Single String
 def full_prediction(sequence):
     features = extract_features(sequence)
-    if isinstance(features, str):
+    if isinstance(features, str):  # error message returned
         return features
     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_values = predictmic(sequence)
-    result = f"Prediction: {amp_result}\nConfidence: {confidence}%\n\nPredicted MIC Values (µM):\n"
-    for organism, mic in mic_values.items():
-        result += f"- {organism}: {mic}\n"
+    
+    result = f"Prediction: {amp_result}\nConfidence: {confidence}%\n"
+    
+    if prediction == 0:  # only predict MIC if AMP
+        mic_values = predictmic(sequence)
+        result += "\nPredicted MIC Values (µM):\n"
+        for organism, mic in mic_values.items():
+            result += f"- {organism}: {mic}\n"
+    else:
+        result += "\nMIC prediction is not available because sequence is Non-AMP."
+    
     return result
 
+
 # Gradio Interface (Single Label Output)
 iface = gr.Interface(
     fn=full_prediction,