Update app.py

app.py

@@ -167,7 +167,7 @@ def submit_result(name, age, medical_record, sex, input_image, predicted_image,
     conn.close()
     return "Result submitted to database."
 
-# Function to load and view database
+# Function to load and view database in HTML format
 def view_database():
     conn = sqlite3.connect('results.db')
     c = conn.cursor()
@@ -175,16 +175,17 @@ def view_database():
     rows = c.fetchall()
     conn.close()
 
-    # Prepare the data for display
-    data = []
+    # Prepare the HTML content
+    html_content = "<table border='1'><tr><th>Name</th><th>Age</th><th>Medical Record</th><th>Sex</th><th>Input Image</th><th>Predicted Image</th><th>Result</th></tr>"
+
     for row in rows:
         name, age, medical_record, sex, input_image_bytes, predicted_image_bytes, result = row
-        
+
         # Decode the images
         input_image = Image.open(BytesIO(input_image_bytes))
         predicted_image = Image.open(BytesIO(predicted_image_bytes))
 
-        # Convert images to base64 for display
+        # Convert images to base64 for display in HTML
         buffered_input = BytesIO()
         input_image.save(buffered_input, format="PNG")
         input_image_base64 = base64.b64encode(buffered_input.getvalue()).decode('utf-8')
@@ -193,13 +194,12 @@ def view_database():
         predicted_image.save(buffered_predicted, format="PNG")
         predicted_image_base64 = base64.b64encode(buffered_predicted.getvalue()).decode('utf-8')
 
-        # Append to data
-        data.append([name, age, medical_record, sex, input_image_base64, predicted_image_base64, result])
+        # Add a row to the HTML table
+        html_content += f"<tr><td>{name}</td><td>{age}</td><td>{medical_record}</td><td>{sex}</td><td><img src='data:image/png;base64,{input_image_base64}' width='100'></td><td><img src='data:image/png;base64,{predicted_image_base64}' width='100'></td><td>{result}</td></tr>"
 
-    # Create a DataFrame
-    df = pd.DataFrame(data, columns=["Name", "Age", "Medical Record", "Sex", "Input Image", "Predicted Image", "Result"])
+    html_content += "</table>"
 
-    return df
+    return html_content
 
 # Function to download database or image
 def download_file(choice):
@@ -232,10 +232,10 @@ def interface(name, age, medical_record, sex, input_image):
 
     return output_image, raw_result, submit_status
 
-# View Database Function
+# View Database Function (Updated)
 def view_db_interface():
-    df = view_database()
-    return df
+    html_content = view_database()
+    return html_content
 
 # Download Function
 def download_interface(choice):
@@ -272,7 +272,7 @@ with gr.Blocks() as demo:
 
     with gr.Column():
         view_db_btn = gr.Button("View Database")
-        db_output = gr.Dataframe(label="Database Records")
+        db_output = gr.HTML(label="Database Records")
 
     view_db_btn.click(fn=view_db_interface, inputs=[], outputs=[db_output])