Update app.py

app.py

@@ -21,14 +21,13 @@ def load_drug_interaction_dataset():
         df = pd.read_csv(dataset_path)
         print(f"Dataset columns: {df.columns.tolist()}")
         print(f"Dataset shape: {df.shape}")
-        print(f"First few rows:\n{df.head().to_string()}")
+        print(f"First few rows:\n{df.head()}")
         
         # Create interaction dictionary using your exact column names
         interaction_db = {}
         count = 0
-        severe_count = 0  # Track severe interactions
         
-        for index, row in df.iterrows():
+        for _, row in df.iterrows():
             try:
                 # Use your exact column names
                 drug1 = str(row['Drug 1_normalized']).lower().strip()
@@ -39,7 +38,6 @@ def load_drug_interaction_dataset():
                 if (not all([drug1, drug2, severity]) or 
                     drug1 == 'nan' or drug2 == 'nan' or 
                     severity == 'nan' or severity.lower() == 'none'):
-                    print(f"Skipping invalid row {index}: {drug1}, {drug2}, {severity}")
                     continue
                 
                 # Clean up severity labels
@@ -49,19 +47,14 @@ def load_drug_interaction_dataset():
                 
                 # Add both orders to the dictionary
                 interaction_db[(drug1, drug2)] = severity
-                interaction_db[(drug2, drug1)] = severity
+                interaction_db[(drug2, drug1)] = severity  # Add reverse order
                 count += 1
-                if severity.lower() == 'severe':
-                    severe_count += 1
-                if drug1 == 'warfarin' and drug2 == 'aspirin':
-                    print(f"Found Warfarin, Aspirin at row {index} with severity: {severity}")
                 
             except Exception as e:
-                print(f"Error processing row {index}: {e}")
+                print(f"Error processing row {_}: {e}")
                 continue
         
         print(f"✅ Successfully loaded {count} drug interactions from dataset")
-        print(f"Number of 'Severe' interactions: {severe_count}")
         print(f"Sample interactions: {list(interaction_db.items())[:5]}")
         return interaction_db
         
@@ -79,12 +72,86 @@ def create_fallback_database():
         ('ibuprofen', 'warfarin'): 'Severe',
         ('simvastatin', 'clarithromycin'): 'Severe',
         ('clarithromycin', 'simvastatin'): 'Severe',
+        ('clopidogrel', 'omeprazole'): 'Severe',
+        ('omeprazole', 'clopidogrel'): 'Severe',
+        ('methotrexate', 'naproxen'): 'Severe',
+        ('naproxen', 'methotrexate'): 'Severe',
+        ('lithium', 'ibuprofen'): 'Severe',
+        ('ibuprofen', 'lithium'): 'Severe',
+        ('ssri', 'maoi'): 'Severe',
+        ('maoi', 'ssri'): 'Severe',
+        ('simvastatin', 'verapamil'): 'Severe',
+        ('verapamil', 'simvastatin'): 'Severe',
+        ('warfarin', 'fluconazole'): 'Severe',
+        ('fluconazole', 'warfarin'): 'Severe',
+        ('digoxin', 'verapamil'): 'Severe',
+        ('verapamil', 'digoxin'): 'Severe',
+        
+        # Moderate interactions
         ('digoxin', 'quinine'): 'Moderate',
         ('quinine', 'digoxin'): 'Moderate',
+        ('lisinopril', 'ibuprofen'): 'Moderate',
+        ('ibuprofen', 'lisinopril'): 'Moderate',
+        ('metformin', 'alcohol'): 'Moderate',
+        ('alcohol', 'metformin'): 'Moderate',
+        ('levothyroxine', 'calcium'): 'Moderate',
+        ('calcium', 'levothyroxine'): 'Moderate',
+        ('atorvastatin', 'orange juice'): 'Moderate',
+        ('orange juice', 'atorvastatin'): 'Moderate',
+        ('phenytoin', 'warfarin'): 'Moderate',
+        ('warfarin', 'phenytoin'): 'Moderate',
+        ('theophylline', 'ciprofloxacin'): 'Moderate',
+        ('ciprofloxacin', 'theophylline'): 'Moderate',
+        ('warfarin', 'acetaminophen'): 'Moderate',
+        ('acetaminophen', 'warfarin'): 'Moderate',
+        ('metoprolol', 'verapamil'): 'Moderate',
+        ('verapamil', 'metoprolol'): 'Moderate',
+        ('spironolactone', 'digoxin'): 'Moderate',
+        ('digoxin', 'spironolactone'): 'Moderate',
+        
+        # Mild interactions
         ('metformin', 'ibuprofen'): 'Mild',
         ('ibuprofen', 'metformin'): 'Mild',
+        ('omeprazole', 'calcium'): 'Mild',
+        ('calcium', 'omeprazole'): 'Mild',
+        ('vitamin d', 'calcium'): 'Mild',
+        ('calcium', 'vitamin d'): 'Mild',
+        ('aspirin', 'vitamin c'): 'Mild',
+        ('vitamin c', 'aspirin'): 'Mild',
+        ('atorvastatin', 'vitamin d'): 'Mild',
+        ('vitamin d', 'atorvastatin'): 'Mild',
+        ('metformin', 'vitamin b12'): 'Mild',
+        ('vitamin b12', 'metformin'): 'Mild',
+        ('omeprazole', 'vitamin b12'): 'Mild',
+        ('vitamin b12', 'omeprazole'): 'Mild',
+        ('aspirin', 'ginger'): 'Mild',
+        ('ginger', 'aspirin'): 'Mild',
+        ('warfarin', 'green tea'): 'Mild',
+        ('green tea', 'warfarin'): 'Mild',
+        ('levothyroxine', 'iron'): 'Mild',
+        ('iron', 'levothyroxine'): 'Mild',
+        
+        # No interactions
         ('vitamin c', 'vitamin d'): 'No Interaction',
         ('vitamin d', 'vitamin c'): 'No Interaction',
+        ('calcium', 'vitamin d'): 'No Interaction',
+        ('vitamin d', 'calcium'): 'No Interaction',
+        ('omeprazole', 'vitamin d'): 'No Interaction',
+        ('vitamin d', 'omeprazole'): 'No Interaction',
+        ('metformin', 'vitamin d'): 'No Interaction',
+        ('vitamin d', 'metformin'): 'No Interaction',
+        ('aspirin', 'vitamin e'): 'No Interaction',
+        ('vitamin e', 'aspirin'): 'No Interaction',
+        ('atorvastatin', 'coenzyme q10'): 'No Interaction',
+        ('coenzyme q10', 'atorvastatin'): 'No Interaction',
+        ('levothyroxine', 'vitamin d'): 'No Interaction',
+        ('vitamin d', 'levothyroxine'): 'No Interaction',
+        ('metoprolol', 'magnesium'): 'No Interaction',
+        ('magnesium', 'metoprolol'): 'No Interaction',
+        ('lisinopril', 'potassium'): 'No Interaction',
+        ('potassium', 'lisinopril'): 'No Interaction',
+        ('simvastatin', 'vitamin e'): 'No Interaction',
+        ('vitamin e', 'simvastatin'): 'No Interaction',
     }
 
 # Load your dataset
@@ -121,14 +188,8 @@ def predict_interaction(drug_names):
             else:
                 return f"📊 **INTERACTION LEVEL**: {prediction}"
         else:
-            # Try to find similar drugs for debugging
-            found_drugs = set()
-            for d1, d2 in interaction_db.keys():
-                found_drugs.add(d1)
-                found_drugs.add(d2)
-            
-            print(f"Not found. Available drugs: {sorted(list(found_drugs))[:20]}...")
-            print(f"Keys in interaction_db: {list(interaction_db.keys())[:5]}... (total {len(interaction_db)})")
+            # Log all available keys for debugging
+            print(f"Keys in interaction_db: {list(interaction_db.keys())[:10]}... (total {len(interaction_db)})")
             
             # Check if either drug exists in the database
             drug1_exists = any(d1 == drug1 or d2 == drug1 for d1, d2 in interaction_db.keys())