Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from service import GPT_Service, Stage | |
| title_html = """ | |
| <div style="display: flex; align-items: center; width=100%; background-color:#FFC90E; border-radius:10px;"> | |
| <img src='https://i.ibb.co/gPFskvg/LOGO-1.png' style='height: 90px; width:90px; margin-left: 20px;'> | |
| <span style="margin-left: 20px;font-size:30px"> | |
| <span>Patient Assessment Documentation</span> | |
| </span> | |
| </div> | |
| """ | |
| css = """ | |
| #subButton, #rephraseBtn { | |
| background-color: #EA580C !important; | |
| background: #EA580C !important; | |
| color: #FFFFFF !important; | |
| border: none !important; /* Remove the border */ | |
| transition: background-color 0.3s ease; /* Add transition for smoother hover effect */ | |
| } | |
| #subButton:hover, #rephraseBtn:hover { | |
| color: #EA580C !important; | |
| background-color: #FFC90E !important; /* Change to a lighter background color on hover */ | |
| } | |
| """ | |
| def split_based_on_commas(string): | |
| return [substring.strip() for substring in string.split(",")] | |
| def process_text(keywords, stage, medical_history): | |
| if(keywords.strip()!='' and stage!=None and len(stage)!=0 and len(split_based_on_commas(keywords))<=5): | |
| try: | |
| gptService=GPT_Service() | |
| if stage == Stage.INITIAL_ASSESSMENT.value: | |
| options = gptService.getInitialAssessment(keywords,medical_history) | |
| return gr.update(choices=options, value="", interactive=True) | |
| elif stage == Stage.FOLLOWUP_ASSESSMENT.value: | |
| options = gptService.getFollowUpAssessment(keywords,medical_history) | |
| return gr.update(choices=options, value="", interactive=True) | |
| elif stage == Stage.EVALUATION.value: | |
| options = gptService.getEvaluation(keywords,medical_history) | |
| return gr.update(choices=options, value="", interactive=True) | |
| elif stage == Stage.DETAILED_EVALUATION.value: | |
| options = gptService.getDetailedEvaluation(keywords,medical_history) | |
| return gr.update(choices=options, value="", interactive=True) | |
| elif stage == Stage.PROGRESS_NOTE.value: | |
| options = gptService.getProgressNote(keywords,medical_history) | |
| return gr.update(choices=options, value="", interactive=True) | |
| elif stage == Stage.DISCHARGE_SUMMARY.value: | |
| options = gptService.getDischargeSummary(keywords,medical_history) | |
| return gr.update(choices=options, value="", interactive=True) | |
| elif stage == Stage.SHORT_TERM_GOALS.value: | |
| options = gptService.getShortTermGoals(keywords,medical_history) | |
| return gr.update(choices=options, value="", interactive=True) | |
| elif stage == Stage.LONG_TERM_GOALS.value: | |
| options = gptService.getLongTermGoals(keywords,medical_history) | |
| return gr.update(choices=options, value="", interactive=True) | |
| elif stage == Stage.RECOMMENDATION.value: | |
| options = gptService.getRecommendation(keywords,medical_history) | |
| return gr.update(choices=options, value="", interactive=True) | |
| else: | |
| print("Unknown stage") | |
| raise gr.Error('Unknown stage encountered') | |
| except: | |
| print("GPT error encounters") | |
| raise gr.Error('Something went wrong please try again') | |
| else: | |
| print("Unknown stage parameters") | |
| raise gr.Error('Provide appropriate parameters and try again') | |
| def clear_input(response): | |
| return gr.update(value="") | |
| def getPatientAssessment(sentences,stage, medical_history): | |
| if(sentences is None or len(sentences) == 0 or stage is None): | |
| raise gr.Error('Provide appropriate parameters and try again') | |
| else: | |
| try: | |
| gptService=GPT_Service() | |
| if stage == Stage.INITIAL_ASSESSMENT.value: | |
| return gptService.getDetailInitialAssessment(medical_history,sentences) | |
| elif stage == Stage.SHORT_TERM_GOALS.value: | |
| return gptService.getDetailShortTermGoals(medical_history,sentences) | |
| elif stage == Stage.LONG_TERM_GOALS.value: | |
| return gptService.getDetailLongTermGoals(medical_history,sentences) | |
| elif stage == Stage.RECOMMENDATION.value: | |
| return gptService.getDetailRecommendation(medical_history,sentences) | |
| else: | |
| print("Unknown stage") | |
| raise gr.Error('Unknown stage encountered') | |
| except: | |
| print("GPT error encounters") | |
| raise gr.Error('Something went wrong please try again') | |
| def getRephrasedSentences(sentences): | |
| if (sentences is None or sentences.strip() == ''): | |
| raise gr.Error('Provide appropriate parameters and try again') | |
| else: | |
| try: | |
| gptService=GPT_Service() | |
| return gptService.getRephrasedPatientAssessment(sentences) | |
| except: | |
| raise gr.Error('Something went wrong please try again') | |
| with gr.Blocks(css=css) as demo: | |
| gr.HTML(title_html) | |
| gr.HTML("<h2>Step 1:- Patient Assessment Input</h2>") | |
| with gr.Row(): | |
| keywords = gr.Textbox(label="Keywords", placeholder="Write comma separated sample keywords (max=5)") | |
| medicalHistory = gr.Textbox(label="Medical History", placeholder="Enter patient history") | |
| stage_choices = [Stage.INITIAL_ASSESSMENT.value,Stage.SHORT_TERM_GOALS.value,Stage.LONG_TERM_GOALS.value, Stage.RECOMMENDATION.value] | |
| stage = gr.Dropdown(choices=stage_choices, label="Assessment Stage") | |
| with gr.Row(): | |
| clearInputFields = gr.ClearButton(value="Clear Inputs",components=[keywords,medicalHistory,stage]) | |
| responseButton = gr.Button("Submit",elem_id="subButton") | |
| with gr.Column(scale=2): | |
| gr.HTML("<h2>Step 2:- Keyword-Driven Sentence Analysis</h2>") | |
| selectedSentence = gr.Dropdown(choices=[], show_label=False,interactive=False,multiselect=True) | |
| with gr.Row(): | |
| clearSelectedSentences = gr.Button(value="Clear Selection") | |
| detailAssessmentButton = gr.Button("Generate Summary",elem_id="rephraseBtn",) | |
| with gr.Column(): | |
| gr.HTML("<h2>Step 3:- Patient Assessment Summary</h2>") | |
| patientDetailAssessment=gr.Textbox(show_label=False, interactive=False, show_copy_button=True,lines=15) | |
| rephraseButton = gr.Button("I don't like response, Generate New :)",elem_id="rephraseBtn",) | |
| responseButton.click(process_text,[keywords,stage,medicalHistory],[selectedSentence]) | |
| detailAssessmentButton.click(getPatientAssessment,[selectedSentence,stage,medicalHistory],patientDetailAssessment) | |
| rephraseButton.click(getPatientAssessment,[selectedSentence,stage,medicalHistory],patientDetailAssessment) | |
| clearSelectedSentences.click(clear_input,inputs=[selectedSentence],outputs=[selectedSentence]) | |
| demo.launch(share=True) |