Update app.py

app.py

@@ -157,7 +157,6 @@ def create_fallback_database():
         ('simvastatin', 'vitamin e'): 'No Interaction',
         ('vitamin e', 'simvastatin'): 'No Interaction',
     }
-    
 
 # Function to fetch drug features from PubChem if available
 def get_pubchem_features(drug_name):
@@ -199,15 +198,26 @@ def predict_interaction(drug_names, dataset_db, fallback_db):
     """Predict interaction between two drugs using dataset, fallback, and PubChem"""
     try:
         if not drug_names or ',' not in drug_names:
-            return "No Interaction"
+            return "Error"
         
         drugs = [drug.strip() for drug in drug_names.split(',')]
         if len(drugs) != 2:
-            return "No Interaction"
+            return "Error"
         
         drug1, drug2 = drugs[0].lower(), drugs[1].lower()
         print(f"Looking up: '{drug1}' + '{drug2}'")
-        
+
+        # Validate drug names
+        all_known_drugs = set()
+        for d1, d2 in {**dataset_db, **fallback_db}.keys():
+            all_known_drugs.add(d1)
+            all_known_drugs.add(d2)
+
+        if (len(drug1) < 2 or len(drug2) < 2 or  # Minimum length check
+            drug1 not in all_known_drugs and not PUBCHEM_AVAILABLE or
+            drug2 not in all_known_drugs and not PUBCHEM_AVAILABLE):
+            return "Error"
+
         # Check dataset first
         prediction = dataset_db.get((drug1, drug2))
         if prediction:
@@ -228,7 +238,7 @@ def predict_interaction(drug_names, dataset_db, fallback_db):
             return random.choice(['Moderate', 'Mild'])
         
     except Exception:
-        return "No Interaction"
+        return "Error"
 
 # Load your dataset and fallback
 print("Loading drug interaction dataset...")