Update app.py

app.py

@@ -6,9 +6,8 @@ from itertools import product
 import torch.nn as nn
 import matplotlib.pyplot as plt
 import matplotlib.colors as mcolors
-import io
-from PIL import Image
-from scipy.interpolate import interp1d
+from io import BytesIO
+from PIL import Image, ImageDraw
 from Bio.Graphics import GenomeDiagram
 from Bio.SeqFeature import SeqFeature, FeatureLocation
 from reportlab.lib import colors
@@ -670,6 +669,12 @@ def analyze_gene_features(sequence_file, features_file, fasta_text="", features_
 
 def create_genome_diagram(gene_results, genome_length):
     """Create genome diagram using BioPython"""
+    from Bio.Graphics import GenomeDiagram
+    from Bio.SeqFeature import SeqFeature, FeatureLocation
+    from reportlab.lib import colors
+    from io import BytesIO
+    from PIL import Image
+    import io  # Ensure io is imported at the top level
     
     # Create diagram
     gd_diagram = GenomeDiagram.Diagram("Genome SHAP Analysis")
@@ -700,21 +705,29 @@ def create_genome_diagram(gene_results, genome_length):
             name=f"{gene['gene_name']}\n(SHAP: {gene['avg_shap']:.3f})"
         )
         
-    # Draw diagram
-    gd_diagram.draw(
-        format="linear",
-        orientation="landscape",
-        pagesize=(15, 5),
-        start=0,
-        end=genome_length,
-        fragments=1
-    )
-    
-    # Save to BytesIO and convert to PIL Image
-    buffer = BytesIO()
-    gd_diagram.write(buffer, "PNG")
-    buffer.seek(0)
-    return Image.open(buffer)
+    try:
+        # Draw diagram
+        gd_diagram.draw(
+            format="linear",
+            orientation="landscape",
+            pagesize=(15, 5),
+            start=0,
+            end=genome_length,
+            fragments=1
+        )
+        
+        # Save to BytesIO and convert to PIL Image
+        buffer = io.BytesIO()
+        gd_diagram.write(buffer, "PNG")
+        buffer.seek(0)
+        return Image.open(buffer)
+    except Exception as e:
+        print(f"Error creating genome diagram: {str(e)}")
+        # Create a simple error image
+        error_img = Image.new('RGB', (800, 100), color='white')
+        draw = ImageDraw.Draw(error_img)
+        draw.text((10, 40), f"Error creating genome diagram: {str(e)}", fill='black')
+        return error_img
 
 ###############################################################################
 # 12. DOWNLOAD FUNCTIONS