Update app.py

app.py

@@ -40,7 +40,17 @@ def get_protein_image(pdbview):
     pdbview.zoom(0.8)
     pdbview.render()
     png = pdbview.pngImage()
-    png_bytes = png.tostring()  # Convert view to bytes
+    print(f"Type of png: {type(png)}")
+    print(f"Dir of png: {dir(png)}")
+    
+    # Try to convert to bytes if possible
+    if 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')
 
 def perform_blast_analysis(sequence):