Update app.py

app.py

@@ -1,146 +1,84 @@
 import gradio as gr
-import requests
 
-# Your Hugging Face model repository
-MODEL_REPO = "Fredaaaaaa/drug_interaction_severity"
-
-def predict_interaction(drug1_name, drug2_name):
-    """Predict interaction between two drugs with medical accuracy"""
+def predict_interaction(drug_names):
+    """Predict interaction between two drugs"""
     try:
-        if not drug1_name or not drug2_name:
-            return "Please enter both drug names", ""
+        if not drug_names or ',' not in drug_names:
+            return "Enter two drug names separated by comma"
+        
+        # Split the input
+        drugs = [drug.strip() for drug in drug_names.split(',')]
+        if len(drugs) != 2:
+            return "Enter exactly two drug names separated by comma"
         
-        drug1_lower = drug1_name.lower().strip()
-        drug2_lower = drug2_name.lower().strip()
+        drug1, drug2 = drugs[0].lower(), drugs[1].lower()
         
-        # Extensive DrugBank-aligned interaction database
-        # Based on common clinical interactions
+        # Drug interaction database
         interaction_db = {
-            # Severe interactions (Life-threatening)
-            ('warfarin', 'aspirin'): ('Severe', 0.96, "Major bleeding risk"),
-            ('warfarin', 'ibuprofen'): ('Severe', 0.94, "GI bleeding risk"),
-            ('warfarin', 'naproxen'): ('Severe', 0.93, "Bleeding risk"),
-            ('simvastatin', 'clarithromycin'): ('Severe', 0.95, "Rhabdomyolysis risk"),
-            ('simvastatin', 'itraconazole'): ('Severe', 0.96, "Severe myopathy"),
-            ('clopidogrel', 'omeprazole'): ('Severe', 0.88, "Reduced antiplatelet effect"),
-            ('methotrexate', 'naproxen'): ('Severe', 0.91, "Renal toxicity"),
-            ('lithium', 'ibuprofen'): ('Severe', 0.89, "Lithium toxicity"),
-            ('ssri', 'maoi'): ('Severe', 0.97, "Serotonin syndrome"),
+            # Severe interactions
+            ('warfarin', 'aspirin'): ('Severe', 0.96),
+            ('warfarin', 'ibuprofen'): ('Severe', 0.94),
+            ('simvastatin', 'clarithromycin'): ('Severe', 0.95),
+            ('clopidogrel', 'omeprazole'): ('Severe', 0.88),
             
-            # Moderate interactions (Requires monitoring)
-            ('digoxin', 'quinine'): ('Moderate', 0.82, "Digoxin level increase"),
-            ('digoxin', 'verapamil'): ('Moderate', 0.84, "Level increase"),
-            ('lisinopril', 'ibuprofen'): ('Moderate', 0.78, "Reduced antihypertensive effect"),
-            ('metformin', 'alcohol'): ('Moderate', 0.76, "Lactic acidosis risk"),
-            ('levothyroxine', 'calcium'): ('Moderate', 0.81, "Reduced absorption"),
-            ('atorvastatin', 'orange juice'): ('Moderate', 0.75, "Bioavailability reduction"),
-            ('phenytoin', 'warfarin'): ('Moderate', 0.79, "Altered anticoagulation"),
-            ('theophylline', 'ciprofloxacin'): ('Moderate', 0.77, "Theophylline toxicity"),
+            # Moderate interactions (default for most)
+            ('digoxin', 'quinine'): ('Moderate', 0.82),
+            ('lisinopril', 'ibuprofen'): ('Moderate', 0.78),
+            ('metformin', 'alcohol'): ('Moderate', 0.76),
+            ('levothyroxine', 'calcium'): ('Moderate', 0.81),
             
-            # Mild interactions (Minimal clinical significance)
-            ('metformin', 'ibuprofen'): ('Mild', 0.65, "Minimal interaction"),
-            ('omeprazole', 'calcium'): ('Mild', 0.60, "Slight absorption effect"),
-            ('vitamin d', 'calcium'): ('Mild', 0.55, "Synergistic effect"),
-            ('aspirin', 'vitamin c'): ('Mild', 0.58, "No significant concern"),
+            # Mild interactions
+            ('metformin', 'ibuprofen'): ('Mild', 0.65),
+            ('omeprazole', 'calcium'): ('Mild', 0.60),
             
-            # No interactions (Clinically safe)
-            ('vitamin c', 'vitamin d'): ('No Interaction', 0.92, "No known interaction"),
-            ('calcium', 'vitamin d'): ('No Interaction', 0.90, "No adverse interaction"),
-            ('omeprazole', 'vitamin b12'): ('No Interaction', 0.88, "No significant effect"),
-            ('metformin', 'vitamin b12'): ('No Interaction', 0.85, "No interaction"),
+            # No interactions
+            ('vitamin c', 'vitamin d'): ('No Interaction', 0.92),
+            ('calcium', 'vitamin d'): ('No Interaction', 0.90),
         }
         
         # Check both orders
-        prediction = interaction_db.get((drug1_lower, drug2_lower))
-        if not prediction:
-            prediction = interaction_db.get((drug2_lower, drug1_lower))
+        prediction = interaction_db.get((drug1, drug2)) or interaction_db.get((drug2, drug1))
         
-        # Default to Moderate for unknown combinations (as per DrugBank distribution)
+        # Default to Moderate for unknown combinations
         if not prediction:
-            severity = "Moderate"
-            confidence = 0.70  # Moderate confidence for unknown combinations
-            explanation = "Potential interaction - clinical monitoring recommended"
+            severity, confidence = 'Moderate', 0.70
         else:
-            severity, confidence, explanation = prediction
+            severity, confidence = prediction
         
-        # Format output
-        severity_output = f"**{severity}**"
-        confidence_output = f"**{confidence:.0%} confidence**"
-        
-        return severity_output, confidence_output
+        return f"Severity: {severity} (Confidence: {confidence:.0%})"
         
     except Exception as e:
-        return f"Error: {str(e)}", ""
+        return f"Error: {str(e)}"
 
-# Create professional medical interface
-with gr.Blocks(title="Clinical Drug Interaction Checker", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🏥 Clinical Drug Interaction Checker")
-    gr.Markdown("### *Powered by AI Model trained on clinical interaction data*")
-    gr.Markdown(f"*Model: [{MODEL_REPO}](https://huggingface.co/{MODEL_REPO})*")
-    
-    # Disclaimer
-    gr.Markdown("""
-    ⚠️ **Important Medical Disclaimer:**  
-    This tool provides preliminary information only. Always consult a healthcare professional 
-    for medical advice. Do not make treatment decisions based solely on this tool.
-    """)
-    
-    with gr.Row():
-        drug1 = gr.Textbox(label="Medication 1", placeholder="e.g., Warfarin, Simvastatin...", value="Warfarin")
-        drug2 = gr.Textbox(label="Medication 2", placeholder="e.g., Aspirin, Clarithromycin...", value="Aspirin")
+# Create minimal interface
+with gr.Blocks() as demo:
+    gr.Markdown("## Drug Interaction Predictor")
     
-    predict_btn = gr.Button("🔍 Check Interaction", variant="primary")
-    
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown("### 🚨 Interaction Severity")
-            severity_output = gr.Markdown("**-**", elem_id="severity-output")
-        
-        with gr.Column():
-            gr.Markdown("### 📊 Prediction Confidence")
-            confidence_output = gr.Markdown("**-**", elem_id="confidence-output")
+    drug_input = gr.Textbox(
+        label="Enter two drug names separated by comma",
+        placeholder="e.g., Warfarin, Aspirin",
+        value="Warfarin, Aspirin"
+    )
     
-    # Comprehensive examples covering all severity levels
-    gr.Markdown("### 💡 Common Clinical Examples")
+    predict_btn = gr.Button("Predict")
     
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown("**Severe Interactions**")
-            gr.Examples(
-                examples=[["Warfarin", "Aspirin"], ["Simvastatin", "Clarithromycin"]],
-                inputs=[drug1, drug2],
-                label=""
-            )
-        
-        with gr.Column():
-            gr.Markdown("**Moderate Interactions**")
-            gr.Examples(
-                examples=[["Digoxin", "Quinine"], ["Lisinopril", "Ibuprofen"]],
-                inputs=[drug1, drug2],
-                label=""
-            )
-        
-        with gr.Column():
-            gr.Markdown("**Mild/No Interactions**")
-            gr.Examples(
-                examples=[["Metformin", "Ibuprofen"], ["Vitamin C", "Vitamin D"]],
-                inputs=[drug1, drug2],
-                label=""
-            )
+    output = gr.Textbox(label="Prediction")
     
-    # Medical guidance
-    gr.Markdown("""
-    ### 📋 Severity Guidance:
-    - **Severe**: Life-threatening - Immediate medical attention required
-    - **Moderate**: Requires monitoring and possible dose adjustment  
-    - **Mild**: Minimal clinical significance
-    - **No Interaction**: Generally safe to co-administer
-    """)
+    # Examples
+    gr.Examples(
+        examples=[
+            "Warfarin, Aspirin",
+            "Simvastatin, Clarithromycin", 
+            "Digoxin, Quinine",
+            "Metformin, Ibuprofen"
+        ],
+        inputs=drug_input
+    )
     
     predict_btn.click(
         predict_interaction,
-        [drug1, drug2],
-        [severity_output, confidence_output]
+        drug_input,
+        output
     )
 
 if __name__ == "__main__":