Migrate to Gradio

main.py

@@ -1,94 +1,15 @@
-from fasthtml.common import *
+import gradio as gr
 from datasets import load_dataset
 import datetime
 from pbs_data import PBSPublicDataAPIClient
 import os
-from fasthtml_hf import setup_hf_backup
-from fasthtml import FastHTML
 from apscheduler.schedulers.background import BackgroundScheduler
 from apscheduler.triggers.interval import IntervalTrigger
 import atexit
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
-
-
-custom_css = Style("""
-    body {
-        font-family: Arial, sans-serif;
-        line-height: 1.6;
-        color: #333;
-        max-width: 800px;
-        margin: 0 auto;
-        padding: 20px;
-        background-color: #f0f0f0;
-    }
-    h1 {
-        color: #2c3e50;
-        text-align: center;
-        margin-bottom: 30px;
-    }
-    form {
-        background-color: #ffffff;
-        padding: 20px;
-        border-radius: 8px;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-    }
-    label {
-        display: block;
-        margin-bottom: 5px;
-        font-weight: bold;
-        color: #2c3e50;
-    }
-    select, button {
-        width: 100%;
-        padding: 10px;
-        margin-bottom: 15px;
-        border: 1px solid #ddd;
-        border-radius: 4px;
-        background-color: #fff;
-        color: #333;
-    }
-    button {
-        background-color: #3498db;
-        color: white;
-        font-weight: bold;
-        cursor: pointer;
-        transition: background-color 0.3s;
-    }
-    button:hover {
-        background-color: #2980b9;
-    }
-    #results {
-        margin-top: 30px;
-        background-color: #ffffff;
-        padding: 20px;
-        border-radius: 8px;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-    }
-    #results h2 {
-        color: #2c3e50;
-        border-bottom: 2px solid #3498db;
-        padding-bottom: 10px;
-    }
-    #results p {
-        margin-bottom: 10px;
-        color: #333;
-    }
-    #results hr {
-        border: none;
-        border-top: 1px solid #eee;
-        margin: 20px 0;
-    }
-    a {
-        color: #3498db;
-        text-decoration: none;
-    }
-    a:hover {
-        text-decoration: underline;
-    }
-""")
-
 DATASET_NAME = "cmcmaster/rheumatology-biologics-dataset"
+UPDATE_INTERVAL = 1
 
 def load_data():
     try:
@@ -125,16 +46,14 @@ def load_data():
 
 biologics_data = load_data()
 
-app, rt = fast_app()
-
-def search_biologics(drug, brand, formulation, indication, treatment_phase, hospital_type):
-    results = biologics_data['combinations'].filter(
+def search_biologics(drug, brand, formulation, indication, treatment_phase, hospital_type, state):
+    results = state['combinations'].filter(
         lambda x: (not drug or x['drug'] == drug) and
-                  (not brand or x['brand'] == brand) and
-                  (not formulation or x['formulation'] == formulation) and
-                  (not indication or x['indication'] == indication) and
-                  (not treatment_phase or x['treatment_phase'] == treatment_phase) and
-                  (not hospital_type or x['hospital_type'] == hospital_type)
+                 (not brand or x['brand'] == brand) and
+                 (not formulation or x['formulation'] == formulation) and
+                 (not indication or x['indication'] == indication) and
+                 (not treatment_phase or x['treatment_phase'] == treatment_phase) and
+                 (not hospital_type or x['hospital_type'] == hospital_type)
     )
     
     if len(results) == 0:
@@ -143,198 +62,167 @@ def search_biologics(drug, brand, formulation, indication, treatment_phase, hosp
     output = ""
     for item in results:
         output += f"""
-        <div class="result-item">
-            <h2>{item['drug']} ({item['brand']})</h2>
-            <p><strong>PBS Code:</strong> <a href="https://www.pbs.gov.au/medicine/item/{item['pbs_code']}" target="_blank">{item['pbs_code']}</a></p>
-            <p><strong>Formulation:</strong> {item['formulation']}</p>
-            <p><strong>Indication:</strong> {item['indication']}</p>
-            <p><strong>Treatment Phase:</strong> {item['treatment_phase']}</p>
-            <p><strong>Streamlined Code:</strong> {item['streamlined_code'] or 'N/A'}</p>
-            <p><strong>Authority Method:</strong> {item['authority_method']}</p>
-            <p><strong>Online Application:</strong> {'Yes' if item['online_application'] else 'No'}</p>
-            <p><strong>Hospital Type:</strong> {item['hospital_type']}</p>
-            <p><strong>Schedule Year:</strong> {item['schedule_year']}</p>
-            <p><strong>Schedule Month:</strong> {item['schedule_month']}</p>
-        </div>
-        <hr>
-        """
-    
+### {item['drug']} ({item['brand']})
+
+* **PBS Code:** [{item['pbs_code']}](https://www.pbs.gov.au/medicine/item/{item['pbs_code']})
+* **Formulation:** {item['formulation']}
+* **Indication:** {item['indication']}
+* **Treatment Phase:** {item['treatment_phase']}
+* **Streamlined Code:** {item['streamlined_code'] or 'N/A'}
+* **Authority Method:** {item['authority_method']}
+* **Online Application:** {'Yes' if item['online_application'] else 'No'}
+* **Hospital Type:** {item['hospital_type']}
+* **Schedule:** {item['schedule_month']} {item['schedule_year']}
+
+---
+"""
     return output
 
-def update_options(drug, brand, formulation, indication, treatment_phase, hospital_type):
-    filtered = biologics_data['combinations'].filter(
-        lambda x: (not drug or x['drug'] == drug) and
-                  (not brand or x['brand'] == brand) and
-                  (not formulation or x['formulation'] == formulation) and
-                  (not indication or x['indication'] == indication) and
-                  (not treatment_phase or x['treatment_phase'] == treatment_phase) and
-                  (not hospital_type or x['hospital_type'] == hospital_type)
-    )
-    
-    options = {
-        'drugs': sorted(set(filtered['drug'])),
-        'brands': sorted(set(filtered['brand'])),
-        'formulations': sorted(set(filtered['formulation'])),
-        'indications': sorted(set(filtered['indication'])),
-        'treatment_phases': sorted(set(filtered['treatment_phase'])),
-        'hospital_types': sorted(set(filtered['hospital_type']))
-    }
-    
-    return options
-
-@rt('/')
-def get():
-    return Div(
-        custom_css,
-        H1("Biologics Prescriber Helper"),
-        Form(
-            Div(
-                Label("Drug:"),
-                Select(Option("All", value=""), *[Option(drug, value=drug) for drug in biologics_data['drugs']], name="drug", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Brand:"),
-                Select(Option("All", value=""), *[Option(brand, value=brand) for brand in biologics_data['brands']], name="brand", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Formulation:"),
-                Select(Option("All", value=""), *[Option(formulation, value=formulation) for formulation in biologics_data['formulations']], name="formulation", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Indication:"),
-                Select(Option("All", value=""), *[Option(indication, value=indication) for indication in biologics_data['indications']], name="indication", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Treatment Phase:"),
-                Select(Option("All", value=""), *[Option(phase, value=phase) for phase in biologics_data['treatment_phases']], name="treatment_phase", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Hospital Type:"),
-                Select(Option("All", value=""), *[Option(ht, value=ht) for ht in biologics_data['hospital_types']], name="hospital_type", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Button("Search", type="submit"),
-                Button("Reset", hx_get="/reset", hx_target="#options")
-            ),
-            hx_post="/search",
-            hx_target="#results",
-            id="options"
-        ),
-        Div(id="results")
-    )
-
-@rt('/reset')
-def get():
-    return Div(
-        custom_css,
-        Form(
-            Div(
-                Label("Drug:"),
-                Select(Option("All", value=""), *[Option(drug, value=drug) for drug in biologics_data['drugs']], name="drug", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Brand:"),
-                Select(Option("All", value=""), *[Option(brand, value=brand) for brand in biologics_data['brands']], name="brand", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Formulation:"),
-                Select(Option("All", value=""), *[Option(formulation, value=formulation) for formulation in biologics_data['formulations']], name="formulation", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Indication:"),
-                Select(Option("All", value=""), *[Option(indication, value=indication) for indication in biologics_data['indications']], name="indication", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Treatment Phase:"),
-                Select(Option("All", value=""), *[Option(phase, value=phase) for phase in biologics_data['treatment_phases']], name="treatment_phase", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Hospital Type:"),
-                Select(Option("All", value=""), *[Option(ht, value=ht) for ht in biologics_data['hospital_types']], name="hospital_type", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Button("Search", type="submit"),
-                Button("Reset", hx_get="/reset", hx_target="#options")
-            ),
-            hx_post="/search",
-            hx_target="#results",
-            id="options"
+def update_data():
+    # Check the date - if it's the first day of the month then update the data, otherwise
+    if datetime.datetime.now().day == 1:
+        print(f"Updating data at {datetime.datetime.now()}")
+        client = PBSPublicDataAPIClient("2384af7c667342ceb5a736fe29f1dc6b", rate_limit=0.2)
+        try:
+            data = client.fetch_rheumatology_biologics_data()
+            client.save_data_to_hf(data, HF_TOKEN, DATASET_NAME)
+            print("Data updated successfully")
+            global biologics_data
+            biologics_data = load_data()
+        except Exception as e:
+            print(f"An error occurred while updating data: {str(e)}")
+    else:
+        print(f"Not updating data at {datetime.datetime.now()}")
+
+def create_interface():
+    with gr.Blocks(title="Biologics Prescriber Helper") as demo:
+        gr.Markdown("# Biologics Prescriber Helper")
+        
+        # Create session state to store filtered data for each user
+        session_data = gr.State(biologics_data)
+        
+        def update_dropdown_choices(drug, brand, formulation, indication, treatment_phase, hospital_type, state):
+            # Filter the dataset based on current selections
+            filtered = state['combinations'].filter(
+                lambda x: (not drug or x['drug'] == drug) and
+                         (not brand or x['brand'] == brand) and
+                         (not formulation or x['formulation'] == formulation) and
+                         (not indication or x['indication'] == indication) and
+                         (not treatment_phase or x['treatment_phase'] == treatment_phase) and
+                         (not hospital_type or x['hospital_type'] == hospital_type)
+            )
+            
+            # Get unique values for each field from filtered dataset
+            available_options = {
+                'drugs': [""] + sorted(set(filtered['drug'])),
+                'brands': [""] + sorted(set(filtered['brand'])),
+                'formulations': [""] + sorted(set(filtered['formulation'])),
+                'indications': [""] + sorted(set(filtered['indication'])),
+                'treatment_phases': [""] + sorted(set(filtered['treatment_phase'])),
+                'hospital_types': [""] + sorted(set(filtered['hospital_type']))
+            }
+            
+            # Return the choices and current values for each dropdown
+            return (
+                gr.Dropdown(choices=available_options['drugs'], value=drug if drug in available_options['drugs'] else ""),
+                gr.Dropdown(choices=available_options['brands'], value=brand if brand in available_options['brands'] else ""),
+                gr.Dropdown(choices=available_options['formulations'], value=formulation if formulation in available_options['formulations'] else ""),
+                gr.Dropdown(choices=available_options['indications'], value=indication if indication in available_options['indications'] else ""),
+                gr.Dropdown(choices=available_options['treatment_phases'], value=treatment_phase if treatment_phase in available_options['treatment_phases'] else ""),
+                gr.Dropdown(choices=available_options['hospital_types'], value=hospital_type if hospital_type in available_options['hospital_types'] else ""),
+                state  # Return state unchanged
+            )
+
+        with gr.Row():
+            with gr.Column():
+                drug = gr.Dropdown(
+                    choices=[""] + biologics_data['drugs'],
+                    label="Drug",
+                    value="",
+                    interactive=True
+                )
+                brand = gr.Dropdown(
+                    choices=[""] + biologics_data['brands'],
+                    label="Brand",
+                    value="",
+                    interactive=True
+                )
+                formulation = gr.Dropdown(
+                    choices=[""] + biologics_data['formulations'],
+                    label="Formulation",
+                    value="",
+                    interactive=True
+                )
+            
+            with gr.Column():
+                indication = gr.Dropdown(
+                    choices=[""] + biologics_data['indications'],
+                    label="Indication",
+                    value="",
+                    interactive=True
+                )
+                treatment_phase = gr.Dropdown(
+                    choices=[""] + biologics_data['treatment_phases'],
+                    label="Treatment Phase",
+                    value="",
+                    interactive=True
+                )
+                hospital_type = gr.Dropdown(
+                    choices=[""] + biologics_data['hospital_types'],
+                    label="Hospital Type",
+                    value="",
+                    interactive=True
+                )
+
+        with gr.Row():
+            search_btn = gr.Button("Search", variant="primary")
+            clear_btn = gr.Button("Reset")
+
+        results = gr.Markdown()
+
+        def reset_inputs(state):
+            return (*update_dropdown_choices("", "", "", "", "", "", state)[:-1], state)
+
+        # Update dropdowns when any selection changes
+        all_dropdowns = [drug, brand, formulation, indication, treatment_phase, hospital_type]
+        for dropdown in all_dropdowns:
+            dropdown.change(
+                fn=update_dropdown_choices,
+                inputs=[drug, brand, formulation, indication, treatment_phase, hospital_type, session_data],
+                outputs=[drug, brand, formulation, indication, treatment_phase, hospital_type, session_data]
+            )
+
+        search_btn.click(
+            fn=search_biologics,
+            inputs=[drug, brand, formulation, indication, treatment_phase, hospital_type, session_data],
+            outputs=results
         )
-    )
 
-@rt('/update_options')
-def get(drug: str = '', brand: str = '', formulation: str = '', indication: str = '', treatment_phase: str = '', hospital_type: str = ''):
-    options = update_options(drug, brand, formulation, indication, treatment_phase, hospital_type)
-    return Div(
-        custom_css,
-        Form(
-            Div(
-                Label("Drug:"),
-                Select(Option("All", value=""), *[Option(d, value=d, selected=(d == drug)) for d in options['drugs']], name="drug", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Brand:"),
-                Select(Option("All", value=""), *[Option(b, value=b, selected=(b == brand)) for b in options['brands']], name="brand", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Formulation:"),
-                Select(Option("All", value=""), *[Option(f, value=f, selected=(f == formulation)) for f in options['formulations']], name="formulation", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Indication:"),
-                Select(Option("All", value=""), *[Option(i, value=i, selected=(i == indication)) for i in options['indications']], name="indication", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Treatment Phase:"),
-                Select(Option("All", value=""), *[Option(p, value=p, selected=(p == treatment_phase)) for p in options['treatment_phases']], name="treatment_phase", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Label("Hospital Type:"),
-                Select(Option("All", value=""), *[Option(ht, value=ht, selected=(ht == hospital_type)) for ht in options['hospital_types']], name="hospital_type", hx_get="/update_options", hx_target="#options", hx_trigger="change", hx_include="[name='drug'],[name='brand'],[name='formulation'],[name='indication'],[name='treatment_phase'],[name='hospital_type']")
-            ),
-            Div(
-                Button("Search", type="submit"),
-                Button("Reset", hx_get="/reset", hx_target="#options")
-            ),
-            hx_post="/search",
-            hx_target="#results",
-            id="options"
+        clear_btn.click(
+            fn=reset_inputs,
+            inputs=[session_data],
+            outputs=[drug, brand, formulation, indication, treatment_phase, hospital_type, session_data]
         )
-    )
-
-@rt('/search')
-def post(drug: str = '', brand: str = '', formulation: str = '', indication: str = '', treatment_phase: str = '', hospital_type: str = ''):
-    results = search_biologics(drug, brand, formulation, indication, treatment_phase, hospital_type)
-    return results
 
-def update_data():
-    print(f"Updating data at {datetime.datetime.now()}")
-    client = PBSPublicDataAPIClient("2384af7c667342ceb5a736fe29f1dc6b", rate_limit=0.2)
-    try:
-        data = client.fetch_rheumatology_biologics_data()
-        client.save_data_to_hf(data, HF_TOKEN, DATASET_NAME)
-        print("Data updated successfully")
-        global biologics_data
-        biologics_data = load_data()
-    except Exception as e:
-        print(f"An error occurred while updating data: {str(e)}")
-
-# Set up the scheduler
-update_data()
-# Set up the scheduler
-scheduler = BackgroundScheduler()
-scheduler.add_job(
-    func=update_data,
-    trigger=IntervalTrigger(hours=24),
-    id='update_data',
-    name='Update Data',
-    replace_existing=True
-)
-scheduler.start()
-
-# Make sure to shut down the scheduler when the app is terminated
+    return demo
+
+if UPDATE_INTERVAL > 0:
+    # Set up the scheduler
+    update_data()
+    scheduler = BackgroundScheduler()
+    scheduler.add_job(
+        func=update_data,
+        trigger=IntervalTrigger(days=UPDATE_INTERVAL),
+        id='update_data',
+        name='Update Data',
+        replace_existing=True
+    )
+    scheduler.start()
 
-atexit.register(lambda: scheduler.shutdown())
+    # Shutdown scheduler when app terminates
+    atexit.register(lambda: scheduler.shutdown())
 
-setup_hf_backup(app)
-serve()
+# Create and launch the interface
+if __name__ == "__main__":
+    demo = create_interface()
+    demo.launch()