Spaces:

WellnessNexus
/

ThesisCataractLatest

Sleeping

ariankhalfani commited on Sep 14

Commit

9bb0397

•

1 Parent(s): 5e6a1f4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -167,18 +167,34 @@ def add_text_and_watermark(image, name, age, medical_record, sex, label):
 # Function to append patient info and image to HTML database
 def append_patient_info_to_html(name, age, medical_record, sex, result, predicted_image_base64):
     html_entry = f"""
-    <div>
-        <h2>Patient Information</h2>
-        <p><strong>Name:</strong> {name}</p>
-        <p><strong>Age:</strong> {age}</p>
-        <p><strong>Medical Record:</strong> {medical_record}</p>
-        <p><strong>Sex:</strong> {sex}</p>
-        <p><strong>Result:</strong> {result}</p>
-        <p><strong>Predicted Image:</strong></p>
-        <img src="data:image/png;base64,{predicted_image_base64}" alt="Predicted Image" width="300">
-    </div>
-    <hr>
     """
     with open(html_db_file, 'a') as f:

 # Function to append patient info and image to HTML database
 def append_patient_info_to_html(name, age, medical_record, sex, result, predicted_image_base64):
+    # Check if the table header is already present
+    if os.stat(html_db_file).st_size == 0:  # Empty file, need to create the table structure
+        with open(html_db_file, 'a') as f:
+            f.write("""
+            <html>
+            <body>
+            <h1>Patient Prediction Database</h1>
+            <table border="1" style="width:100%; border-collapse: collapse;">
+            <tr>
+                <th>Name</th>
+                <th>Age</th>
+                <th>Medical Record</th>
+                <th>Sex</th>
+                <th>Result</th>
+                <th>Predicted Image</th>
+            </tr>
+            """)
+    # Append patient data as a new row in the table
     html_entry = f"""
+    <tr>
+        <td>{name}</td>
+        <td>{age}</td>
+        <td>{medical_record}</td>
+        <td>{sex}</td>
+        <td>{result}</td>
+        <td><img src="data:image/png;base64,{predicted_image_base64}" alt="Predicted Image" width="150"></td>
+    </tr>
     """
     with open(html_db_file, 'a') as f: