Update app.py

app.py

@@ -43,14 +43,24 @@ def get_protein_image(pdbview):
     pdbview.render()
     png = pdbview.pngImage()
     
-    # Convert to IPython Image
-    img = Image(png)
+    print(f"Type of png: {type(png)}")
+    print(f"Dir of png: {dir(png)}")
     
-    # Save to BytesIO
-    img_io = io.BytesIO()
-    img.save(img_io, format='PNG')
-    img_io.seek(0)
-    png_bytes = img_io.getvalue()
+    if hasattr(png, 'shape'):
+        print(f"Shape of png: {png.shape}")
+    
+    if hasattr(png, 'dtype'):
+        print(f"Dtype of png: {png.dtype}")
+    
+    # Try to convert to bytes
+    if isinstance(png, (bytes, bytearray)):
+        png_bytes = png
+    elif hasattr(png, 'tobytes'):
+        png_bytes = png.tobytes()
+    elif hasattr(png, 'tostring'):
+        png_bytes = png.tostring()
+    else:
+        raise TypeError(f"Unable to convert {type(png)} to bytes")
     
     return base64.b64encode(png_bytes).decode('utf-8')