Update app.py

app.py

@@ -273,7 +273,28 @@ def analyse_query(query):
     else:
         return f"After you answer the patient's query,  if you think it is the right time, ask him the following question like a doctor would normally do :{current_question}"
 
-
+def prepare_patient_info():
+  default_value = ""
+  patient_info = {
+      "name": patient_info1,
+      "age": patient_info2,
+      "gender": patient_info3,
+  }
+  print(patient_info)
+  for i, key in enumerate(["symptoms", "medications", "conditions_allergies", "lifestyle_changes"]):
+      patient_info[key] = answers[i] if i < len(answers) else default_value
+
+  medical_image_path = "lung disease.png"
+  try:
+      patient_info["medical_image"] = base64.b64encode(open(medical_image_path, "rb").read()).decode()
+  except FileNotFoundError:
+      patient_info["medical_image"] = default_value
+
+  patient_info.update({
+      "header_image": base64.b64encode(open("logo1.png", "rb").read()).decode(),
+  })
+  return patient_info
+    
 def generate_response(query, img="", is_there_img=False):
     global memory
     report_question = analyse_query(query)
@@ -519,7 +540,8 @@ def back():
   return gr.update(visible=False),gr.update(visible=True),gr.update(visible=True),gr.update(visible=True),gr.update(visible=False)
 
 def move_next_page(a, b, c, d):
-    patient_info1 = a + b
+    global patient_info1, patient_info2, patient_info3
+    patient_info1 = a + " " + b
     patient_info2 = c
     patient_info3 = d
     return (
@@ -532,13 +554,14 @@ def move_next_page(a, b, c, d):
 
 
 def view_report():
-    return (
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(visible=True),
-    )
+  patient_info = prepare_patient_info()
+  generate_report(patient_info)
+  report_html ="""
+        <embed src="/file=output.pdf" type="application/pdf" width="100%" height="700px" />
+  """
+  print("report is generated")
+  time.sleep(0.5)
+  return gr.update(visible=False),gr.update(visible=False),gr.update(visible=False),gr.update(visible=False),gr.update(visible=True),report_html
 
 
 css = """.gradio-container {
@@ -601,11 +624,9 @@ with gr.Blocks(css=css) as demo:
             pdf_btn = gr.Button("عرض التقرير الطبي", size="sm", variant="primary")
     with gr.Group(visible=False) as pdf_report:
         gr.HTML(not_talking)
-        gr.HTML(
-            """
-        <embed src="output.pdf" type="application/pdf" width="100%" height="700px" />
-  """
-        )
+        report_html = gr.HTML("""
+        <embed src="/file=output.pdf" type="application/pdf" width="100%" height="700px" />
+  """)
         with gr.Row():
             back_btn = gr.Button("العودة إلى المحادثة", size="sm", variant="primary")
 
@@ -640,7 +661,7 @@ with gr.Blocks(css=css) as demo:
             outputs=[page1, page2, chatbot, html_block, pdf_report],
         )
         pdf_btn.click(
-            view_report, outputs=[page1, page2, chatbot, html_block, pdf_report]
+            view_report, outputs=[page1, page2, chatbot, html_block, pdf_report, report_html]
         )
         back_btn.click(
             back, outputs=[page1, page2, chatbot, html_block, pdf_report]