Update app.py

app.py

@@ -63,6 +63,26 @@ def add_protein_features(cds_info):
             cds['Isoelectric Point'] = 'N/A'
     return cds_info
 
+# Function to add genome_diagram
+def create_genome_diagram(genbank_content, output_file_path):
+    record = SeqIO.read(StringIO(genbank_content), "genbank")
+    gd_diagram = GenomeDiagram.Diagram(record.id)
+    gd_track_for_features = gd_diagram.new_track(1, name="Annotated Features")
+    gd_feature_set = gd_track_for_features.new_set()
+
+    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
+        gd_feature_set.add_feature(feature, color=color, label=True, label_size=10, label_angle=0)
+
+    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")
+
+
 # Streamlit UI
 st.set_page_config(page_title="Genomic Data Dashboard", page_icon="🧬", layout="wide")
 uploaded_file = st.file_uploader("Upload a GenBank file", type=['gb', 'gbk'])
@@ -106,6 +126,13 @@ if uploaded_file is not None:
         kmers = calculate_kmers(sequence, k)
         st.bar_chart(pd.DataFrame.from_dict(kmers, orient='index', columns=['Frequency']).sort_values('Frequency', ascending=False).head(20))
 
+        # Generate genome diagram
+        output_file_path = os.path.join(st.session_state.get("temp_dir", "."), "genome_diagram.png")
+        create_genome_diagram(uploaded_file.getvalue().decode("utf-8"), output_file_path)
+
+        # Display genome diagram
+        st.image(output_file_path, caption='Genome Diagram')
+        
     # Additional Information
     with st.expander("View All Genes"):
         st.dataframe(gene_df)