Update app.py

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 from Bio import SeqIO
 from Bio.SeqUtils.ProtParam import ProteinAnalysis
 from Bio.Graphics import GenomeDiagram
-from reportlab.lib import colors
+from reportlab.lib.colors import Color, lightblue, blue
 from reportlab.lib.units import cm
 from io import StringIO
 from collections import Counter
@@ -80,14 +80,12 @@ def create_genome_diagram(genbank_content, output_file_path):
     for feature in record.features:
         if feature.type not in ["gene", "CDS"]:
             continue
-        if len(gd_feature_set) % 2 == 0:
-            color = colors.blue
-        else:
-            color = colors.lightblue
+        color = lightblue if len(gd_feature_set) % 2 == 0 else blue
         gd_feature_set.add_feature(feature, color=color, label=True, label_size=10, label_angle=0)
 
+    # Change the output format to SVG
     gd_diagram.draw(format="circular", circular=True, pagesize=(20*cm, 20*cm), start=0, end=len(record), circle_core=0.7)
-    gd_diagram.write(output_file_path, "PNG")
+    gd_diagram.write(output_file_path, "SVG")
 
 # Streamlit UI setup
 st.set_page_config(page_title="Genomic Data Dashboard", page_icon="🧬", layout="wide")
@@ -133,9 +131,9 @@ if uploaded_file is not None:
         st.bar_chart(pd.DataFrame.from_dict(kmers, orient='index', columns=['Frequency']).sort_values('Frequency', ascending=False).head(20))
 
     # Generate and display genome diagram
-    output_file_path = os.path.join(temp_dir, "genome_diagram.png")
-    create_genome_diagram(uploaded_file.getvalue().decode("utf-8"), output_file_path)
-    st.image(output_file_path, caption='Genome Diagram')
+    output_file_path_svg = os.path.join(temp_dir, "genome_diagram.svg")
+    create_genome_diagram(uploaded_file.getvalue().decode("utf-8"), output_file_path_svg)
+    st.image(output_file_path_svg, caption='Genome Diagram', format='SVG')
 
     # Additional Information
     with st.expander("View All Genes"):