Update app.py

app.py

@@ -173,10 +173,31 @@ def view_database():
     c.execute("SELECT name, age, medical_record, sex, input_image, predicted_image, result FROM results")
     rows = c.fetchall()
     conn.close()
-    
-    # Convert to pandas DataFrame for better display in Gradio
-    df = pd.DataFrame(rows, columns=["Name", "Age", "Medical Record", "Sex", "Input Image", "Predicted Image", "Result"])
-    
+
+    # Prepare the data for display
+    data = []
+    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
+        buffered_input = BytesIO()
+        input_image.save(buffered_input, format="PNG")
+        input_image_base64 = base64.b64encode(buffered_input.getvalue()).decode('utf-8')
+
+        buffered_predicted = BytesIO()
+        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])
+
+    # Create a DataFrame
+    df = pd.DataFrame(data, columns=["Name", "Age", "Medical Record", "Sex", "Input Image", "Predicted Image", "Result"])
+
     return df
 
 # Function to download database or image