Spaces:

WellnessNexus
/

ThesisCataractLatest

Sleeping

ariankhalfani commited on Sep 14

Commit

8b3a389

•

1 Parent(s): 0f7ace5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -165,13 +165,9 @@ def add_text_and_watermark(image, name, age, medical_record, sex, label):
     return image_with_watermark
-# Function to append patient info and image to HTML database
-import os
-# 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>
@@ -192,7 +188,10 @@ def append_patient_info_to_html(name, age, medical_record, sex, result, predicte
             <tbody>
             """)
-    # Append patient data with a horizontal line and descriptive headers
     html_entry = f"""
     <tr>
         <td>{name}</td>
@@ -202,12 +201,21 @@ def append_patient_info_to_html(name, age, medical_record, sex, result, predicte
         <td>{result}</td>
         <td><img src="data:image/png;base64,{predicted_image_base64}" alt="Predicted Image" width="150"></td>
     </tr>
-    <tr><td colspan="6"><hr></td></tr>  <!-- Horizontal line across all columns -->
     """
     with open(html_db_file, 'a') as f:
         f.write(html_entry)
     return str(html_db_file)  # Return the HTML file path for download
 # Function to download the folders

     return image_with_watermark
 def append_patient_info_to_html(name, age, medical_record, sex, result, predicted_image_base64):
+    # Check if the HTML file is empty or if the table structure is missing
+    if os.stat(html_db_file).st_size == 0:  # Empty file, create the table structure
         with open(html_db_file, 'a') as f:
             f.write("""
             <html>
             <tbody>
             """)
+    # Check if this patient already exists to prevent duplicate entries
+    # This can be improved by checking unique identifiers like `medical_record`
+    # Assuming the uniqueness of the medical record
     html_entry = f"""
     <tr>
         <td>{name}</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:
         f.write(html_entry)
+    # Ensure we only add the closing tags once
+    if "</tbody></table></body></html>" not in open(html_db_file).read():
+        with open(html_db_file, 'a') as f:
+            f.write("""
+            </tbody>
+            </table>
+            </body>
+            </html>
+            """)
     return str(html_db_file)  # Return the HTML file path for download
 # Function to download the folders