Update app.py

app.py

@@ -21,13 +21,14 @@ 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()}")
+        print(f"First few rows:\n{df.head().to_string()}")
         
         # Create interaction dictionary using your exact column names
         interaction_db = {}
         count = 0
+        severe_count = 0  # Track severe interactions
         
-        for _, row in df.iterrows():
+        for index, row in df.iterrows():
             try:
                 # Use your exact column names
                 drug1 = str(row['Drug 1_normalized']).lower().strip()
@@ -38,6 +39,7 @@ 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
@@ -45,16 +47,21 @@ def load_drug_interaction_dataset():
                 if severity == 'No interaction':
                     severity = 'No Interaction'
                 
-                # Add both orders to the dictionary - STORE ONLY THE SEVERITY STRING
+                # Add both orders to the dictionary
                 interaction_db[(drug1, drug2)] = severity
-                interaction_db[(drug2, drug1)] = severity  # Add reverse order
+                interaction_db[(drug2, drug1)] = severity
                 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 {_}: {e}")
+                print(f"Error processing row {index}: {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
         
@@ -66,93 +73,18 @@ def create_fallback_database():
     """Fallback database if dataset loading fails - FIXED STRUCTURE"""
     print("Using fallback database")
     return {
-        # Severe interactions (Life-threatening) - STORE ONLY SEVERITY STRINGS
         ('warfarin', 'aspirin'): 'Severe',
         ('aspirin', 'warfarin'): 'Severe',
         ('warfarin', 'ibuprofen'): 'Severe',
         ('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 (Requires monitoring)
         ('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 (Minimal clinical significance)
         ('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 (Clinically safe)
         ('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
@@ -196,6 +128,7 @@ def predict_interaction(drug_names):
                 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)})")
             
             # Check if either drug exists in the database
             drug1_exists = any(d1 == drug1 or d2 == drug1 for d1, d2 in interaction_db.keys())
@@ -259,12 +192,6 @@ with gr.Blocks(title="Drug Interaction Predictor", theme=gr.themes.Soft()) as de
     
     predict_btn.click(predict_interaction, drug_input, output)
     
-    # Debug button functionality
-    def show_debug():
-        return debug_database(), gr.update(visible=True)
-    
-    debug_btn.click(show_debug, outputs=[debug_output, debug_output])
-    
     # Add disclaimer
     gr.Markdown("""
     ---