Update app.py (#9)

- Update app.py (916ca956de3c231596b3c726f69111e009773bf2)


Co-authored-by: Umar <umar-ts@users.noreply.huggingface.co>

app.py

@@ -1,29 +1,67 @@
 import gradio as gr
 from service import GPT_Service, Stage
 
-# Define the function to be called when the button is clicked
+
+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):
+    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:
-                return gptService.getInitialAssessment(keywords,medical_history)
+                options = gptService.getInitialAssessment(keywords,medical_history)
+                return gr.update(choices=options, value="", interactive=True)
             elif stage == Stage.FOLLOWUP_ASSESSMENT.value:
-                return gptService.getFollowUpAssessment(keywords,medical_history)
+                options = gptService.getFollowUpAssessment(keywords,medical_history)
+                return gr.update(choices=options, value="", interactive=True)
             elif stage == Stage.EVALUATION.value:
-                return gptService.getEvaluation(keywords,medical_history)
+                options = gptService.getEvaluation(keywords,medical_history)
+                return gr.update(choices=options, value="", interactive=True)
             elif stage == Stage.DETAILED_EVALUATION.value:
-                return gptService.getDetailedEvaluation(keywords,medical_history)
+                options = gptService.getDetailedEvaluation(keywords,medical_history)
+                return gr.update(choices=options, value="", interactive=True)
             elif stage == Stage.PROGRESS_NOTE.value:
-                return gptService.getProgressNote(keywords,medical_history)
+                options = gptService.getProgressNote(keywords,medical_history)
+                return gr.update(choices=options, value="", interactive=True)
             elif stage == Stage.DISCHARGE_SUMMARY.value:
-                return gptService.getDischargeSummary(keywords,medical_history)
+                options = gptService.getDischargeSummary(keywords,medical_history)
+                return gr.update(choices=options, value="", interactive=True)
             elif stage == Stage.SHORT_TERM_GOALS.value:
-                return gptService.getShortTermGoals(keywords,medical_history)
+                options = gptService.getShortTermGoals(keywords,medical_history)
+                return gr.update(choices=options, value="", interactive=True)
             elif stage == Stage.LONG_TERM_GOALS.value:
-                return gptService.getLongTermGoals(keywords,medical_history)
+                options = gptService.getLongTermGoals(keywords,medical_history)
+                return gr.update(choices=options, value="", interactive=True)
             elif stage == Stage.RECOMMENDATION.value:
-                return gptService.getRecommendation(keywords,medical_history)
+                options =  gptService.getRecommendation(keywords,medical_history)
+                return gr.update(choices=options, value="", interactive=True)
             else:
                 print("Unknown stage")
                 raise gr.Error('Unknown stage encountered')
@@ -33,46 +71,78 @@ def process_text(keywords, stage, medical_history):
     else:
         print("Unknown stage parameters")
         raise gr.Error('Provide appropriate parameters and try again')
-    
-
-# Create text input components with labels
-input_text = gr.Textbox(lines=7, label="Sample keywords", placeholder="Enter sample keywords here")
-medical_history = gr.Textbox(lines=7, label="Medical History", placeholder="Enter medical history here")
-
-# Populating stage choices and creating dropdown component with label
-
-# To add all the stages
-# stage_choices = [stage.value for stage in Stage]
-
-# To add few stages
-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="Stage")
+    
+def clear_input(response):
+    return gr.update(value="")
 
-# Create output containers
-output_text_1 = gr.Textbox(lines=6, label="Variant 1", show_copy_button=True)
-output_text_2 = gr.Textbox(lines=6, label="Variant 2", show_copy_button=True)
-output_text_3 = gr.Textbox(lines=6, label="Variant 3", show_copy_button=True)
 
-title_html = """
-<div style="display: flex; align-items: center; width=100%; background-color:#FFC90E; overflow:auto;">
-    <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 &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</span>
-    </span>
-</div>
+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")
 
-# Create a Gradio interface
-interface=gr.Interface(fn=process_text,
-             inputs=[input_text, stage, medical_history],
-             outputs=[output_text_1,output_text_2,output_text_3],
-             title=title_html,
-             theme=gr.themes.Default(),
-             allow_flagging='never')
 
-interface.launch()
+    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)