Update app.py

app.py

@@ -41,28 +41,16 @@ def render_mol(pdb):
 def get_protein_image(pdbview):
     pdbview.zoom(0.8)
     pdbview.render()
-    png = pdbview.pngImage()
     
-    print(f"Type of png: {type(png)}")
-    print(f"Dir of png: {dir(png)}")
+    # Get the image data directly as a base64 encoded string
+    png_base64 = pdbview.js().png()
     
-    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()
+    # The png_base64 should already be a base64 encoded string
+    # If it's not, we might need to encode it
+    if isinstance(png_base64, str):
+        return png_base64
     else:
-        raise TypeError(f"Unable to convert {type(png)} to bytes")
-    
-    return base64.b64encode(png_bytes).decode('utf-8')
+        return base64.b64encode(png_base64).decode('utf-8')
 
 def perform_blast_analysis(sequence):
     st.subheader('Protein Analysis')