Update app.py

app.py

@@ -3,125 +3,11 @@ import streamlit as st
 import pandas as pd
 from datasets import load_from_disk
 from transformers import AutoTokenizer, TFAutoModel
-from dotenv import load_dotenv
+from constant import DRGUS_STR_LIST
 
-load_dotenv()
-
-# Drugs = os.getenv('DRUGS')
-
-Drugs = ['Amitriptyline', 'Ambien', 'Aripiprazole', 'Actos', 'Aubra',
-       'Amphetamine / dextroamphetamine', 'Afrezza',
-       'Aluminum chloride hexahydrate', 'Acetaminophen / hydrocodone',
-       'Aviane', 'Aldesleukin', 'Acamprosate', 'Azathioprine',
-       'Adapalene / benzoyl peroxide',
-       'Acetaminophen / butalbital / caffeine', 'Accutane',
-       'Ayr Saline Nasal', 'Asenapine', 'Adipex-P', 'Augmentin', 'Apri',
-       'Alesse', 'Aspirin / carisoprodol', 'Azelaic acid', 'Azithromycin',
-       'Alpha 1-proteinase inhibitor', 'Abilify', 'Amlodipine',
-       'Arimidex', 'Acetaminophen / codeine', 'Advair Diskus',
-       'Alprazolam', 'Alcaftadine',
-       'Acetaminophen / dexbrompheniramine / pseudoephedrine', 'AndroGel',
-       'Avinza', 'Ativan', 'Atomoxetine', 'Alphagan P', 'Apremilast',
-       'Advair HFA', 'Anastrozole', 'Azor', 'Azelastine / fluticasone',
-       'Amerge', 'Avonex Pen', 'Acetaminophen / oxycodone',
-       'Abacavir / dolutegravir / lamivudine', 'Adderall', 'Atorvastatin',
-       'Aptensio XR',
-       'Acetaminophen / dichloralphenazone / isometheptene mucate',
-       'Acyclovir', 'Armodafinil', 'Adderall XR',
-       'Antipyrine / benzocaine', 'Aczone', 'Amoxicillin / clavulanate',
-       'Adalimumab', 'Aranesp', 'Ammonium lactate / halobetasol',
-       'Atenolol', 'Avelox', 'Azithromycin Dose Pack', 'Adapalene',
-       'Amoxicillin', 'Acetaminophen / aspirin / caffeine', 'Azelastine',
-       'Albuterol', 'Amethyst', 'Asacol', 'Aleve', 'Alprostadil',
-       'Astelin', 'Atropine / hyoscyamine / phenobarbital / scopolamine','Ascorbic acid', 'Asthmanefrin', 'Allegra-D 12 Hour', 'Amikacin',
-       'Arixtra', 'Apokyn', 'AbobotulinumtoxinA', 'Axid', 'Accolate',
-       'Adoxa', 'Aspirin / caffeine', 'Aliskiren / hydrochlorothiazide',
-       'Afinitor', 'Acetaminophen / phenylephrine', 'Atacand',
-       'Allergy DN PE','Bactrim', 'Belviq', 'Blisovi Fe 1 / 20',
-       'Benzoyl peroxide / clindamycin', 'BuSpar', 'Bupropion',
-       'Bacitracin / neomycin / polymyxin b', 'Brisdelle', 'Beyaz',
-       'Buprenorphine / naloxone', 'Benzonatate', 'Bisacodyl', 'Buprenex',
-       'Benzoyl peroxide / erythromycin', 'Budeprion SR',
-       'Bupropion / naltrexone', 'Belsomra', 'Bevacizumab',
-       'Barium sulfate', 'Bismuth subsalicylate', 'Brovana', 'Buspirone',
-       'Bronkaid', 'Benzocaine', 'Butrans', 'Bunavail', 'Benicar',
-       'Benadryl Allergy', 'Benadryl', 'Biotin', 'Bromfed DM',
-       'Breo Ellipta', 'Biaxin', 'Botox', 'Buprenorphine',
-       'Benzoic acid / salicylic acid', 'Brimonidine / timolol',
-       'Beclomethasone', 'Bydureon', 'Bonine', 'Brilinta', 'Bactrim DS',
-       'Brexpiprazole', 'Baclofen', 'Benazepril', 'Biaxin XL', 'Basaglar',
-       'Bismuth subcitrate potassium / metronidazole / tetracycline',
-       'Belbuca', 'Bromocriptine', 'Budesonide / formoterol',
-       'Blisovi 24 Fe', 'Bystolic', 'Banzel', 'Balacet', 'Balsalazide',
-       'Benlysta', 'Belimumab', 'Byetta',
-       'Bisoprolol / hydrochlorothiazide', 'Butorphanol', 'Boniva',
-       'Briviact', 'Brimonidine', 'Biafine', 'Betaseron',
-       'Benazepril / hydrochlorothiazide', 'Benzoyl peroxide / sulfur',
-       'Bisacodyl / polyethylene glycol 3350 / potassium chloride / sodium bicarbonate / sodium chloride',
-       'Balsam peru / castor oil / trypsin',
-       'Betamethasone / calcipotriene', 'Budeprion XL', 'Benicar HCT',
-       'Bepotastine', 'Bentyl', 'Brompheniramine', 'Benzaclin',
-       'Bisoprolol', 'Budesonide', 'Bontril Slow Release', 'Bepreve','Berinert', 'Benzoyl peroxide', 'Belladonna / opium',
-       'Butabarbital', 'Bioflavonoids / zinc glycinate', 'Brodalumab',
-       'Bicillin L-A', 'Bimatoprost', 'Benztropine', 'Bethanechol',
-       'Bupivacaine', 'Bosutinib', 'Bismuth subgallate', 'BenzEFoam',
-       'Brivaracetam', 'Blistex','Contrave', 'Cyclafem 1 / 35', 'Copper', 'Chantix',
-       'Ciprofloxacin', 'Cyclosporine', 'Clonazepam', 'Ciclopirox',
-       'Campral', 'Cryselle', 'Clindamycin', 'Clonidine', 'Celecoxib',
-       'Caffeine', 'Clarithromycin', 'Clomiphene', 'Clotrimazole',
-       'Celexa', 'Codeine / guaifenesin', 'Cefuroxime', 'Cymbalta',
-       'Canagliflozin', 'Citalopram', 'Corticotropin', 'Cefdinir',
-       'Carvedilol', 'Chlordiazepoxide', 'CellCept',
-       'Cobicistat / elvitegravir / emtricitabine / tenofovir alafenamide',
-       'Chateal', 'Cyclobenzaprine', 'Cialis', 'Cholestyramine', 'Cozaar',
-       'Catapres', 'Carbidopa / levodopa', 'Carisoprodol',
-       'Citric acid / magnesium oxide / sodium picosulfate', 'Cetirizine',
-       'Cambia', 'Clindamycin / tretinoin', 'Crestor', 'Copaxone',
-       'Chlorpheniramine / hydrocodone / pseudoephedrine',
-       'Cobicistat / elvitegravir / emtricitabine / tenofovir',
-       'Cogentin', 'Cosentyx', 'Cipro', 'Cytomel', 'Camrese', 'Clomid',
-       'Celebrex', 'Colesevelam', 'Clozapine', 'Cutivate', 'Cyred',
-       'Concerta', 'Chlorpheniramine / hydrocodone', 'Claravis',
-       'Cyanocobalamin', 'Clopidogrel', 'Cyproheptadine',
-       'Chlordiazepoxide / clidinium', 'Carbamazepine', 'Crisaborole',
-       'Colchicine', 'Ciprofloxacin / dexamethasone', 'Clomipramine',
-       'Cabergoline', 'Carac', 'Cephalexin', 'Cariprazine', 'Correctol',
-       'Celestone', 'Creon', 'Clobetasol', 'Colazal', 'Chlorzoxazone',
-       'Cenestin', 'Casodex', 'Cinryze', 'Claritin-D', 'Chlorhexidine',
-       'Complera', 'Cervidil', 'Cefixime', 'Coreg', 'Camila',
-       'Clorazepate', 'Cevimeline', 'Cosopt', 'Chondroitin / glucosamine','Citrate of Magnesia', 'Coricidin HBP Cold & Flu', 'Cefditoren',
-       'Ceftibuten', 'Cyclessa', 'Cortef', 'Calcium acetate', 'Cyclizine',
-       'Coagulation factor ix', 'Colace', 'Carmol 20', 'Calan SR',
-       'Cyklokapron', 'Coal tar', 'Cobicistat / darunavir', 'Calamine',
-       'CoQ10','Duloxetine', 'Depakote', 'Drospirenone / estradiol',
-       'Depo-Provera', 'Desyrel', 'Desvenlafaxine',
-       'Drospirenone / ethinyl estradiol', 'Doxylamine / pyridoxine',
-       'Demerol', 'Dextromethorphan', 'Diazepam', 'Diphenhydramine',
-       'Denosumab', 'Dulaglutide', 'Drysol', 'Divalproex sodium',
-       'Doxycycline', 'Desogestrel / ethinyl estradiol', 'Duofilm',
-       'Dicyclomine', 'Dexmethylphenidate', 'Diltiazem', 'Dapsone',
-       'Dalfampridine', 'Dilantin', 'Dienogest / estradiol', 'Diclofenac',
-       'Donnatal', 'Depakote ER', 'Donepezil', 'Dulcolax', 'Dulera',
-       'Dapagliflozin', 'Duexis', 'Differin', 'Doxepin', 'Docosanol',
-       'Diclegis', 'Desloratadine',
-       'Drospirenone / ethinyl estradiol / levomefolate calcium', 'Duac',
-       'Deplin', 'Doryx', 'Dilaudid', 'Dimenhydrinate', 'Delsym',
-       'Denavir', 'D.H.E. 45', 'Disulfiram', 'Droperidol', 'Dasatinib',
-       'Dextrostat', 'Dymista', 'Dextroamphetamine', 'DDAVP Rhinal Tube',
-       'Dabigatran', 'Dasabuvir / ombitasvir / paritaprevir / ritonavir',
-       'Dextromethorphan / guaifenesin', 'Diflucan', 'Debrox',
-       'Diphenhydramine / naproxen', 'Daklinza', 'Daliresp',
-       'Dihydroergotamine', 'Dinoprostone', 'Dermal filler', 'Doxylamine',
-       'Daytrana', 'Diprivan', 'Dexlansoprazole', 'Dovonex', 'Doral',
-       'Desquam-X Wash', 'Dexilant', 'Dofetilide', 'Diovan HCT', 'Detrol',
-       "Dimetapp Children's Cold & Cough", 'Delatestryl', 'Desipramine',
-       'Daclatasvir', 'Depo-Testosterone', 'Dulcolax Laxative',
-       'Dexamethasone', 'Dimethyl fumarate', 'Dronabinol', 'Duragesic',
-       'Dexedrine', 'Dupixent', 'Dramamine','Daypro', 'Dyazide', 'Deltasone', 'Depo-Medrol',
-       'Dapagliflozin / metformin', 'Dilaudid-HP', 'Doxorubicin',
-       'Deoxycholic acid',
-       'Dextromethorphan / phenylephrine / pyrilamine',
-       'Diphenhydramine / ibuprofen', 'Divigel', 'Dermatop']
+if DRGUS_STR_LIST:
+    Drugs = DRGUS_STR_LIST.split(',')
+    Drugs = [drug.strip() for drug in Drugs] 
 
 model_ckpt = "sentence-transformers/multi-qa-mpnet-base-dot-v1"
 tokenizer = AutoTokenizer.from_pretrained(model_ckpt)