Fix code for multiple sequence input

gc_calc.py

@@ -7,7 +7,17 @@
 
 # ===================================================== FUNCTIONS =================================================================
 def calculate_gc(sequence):
-    print(gc_fraction(sequence))
+    print(round(gc_fraction(sequence), 3))
+
+def multiple_gc(seq_choice):
+    sequences = []
+    gc_list = []
+    print("All sequences must be of the same length.")
+    for i in range(seq_choice):
+        sequence = Seq(input("Please enter a DNA/RNA sequence. "))
+        sequences.append(f"Seq {i+1}")
+        gc_list.append(round(gc_fraction(sequence), 3))
+    return sequences , gc_list
 
 def calculate_window_gc(sequence, window, step):
     gc_parts = []
@@ -20,27 +30,87 @@ def calculate_window_gc(sequence, window, step):
         window_list.append(f"Window {window_count}")
     return window_list, gc_parts
 
-def create_window_plot():
+
+def multi_window_plot(x_windows, seq_dict):
+    colors_list = ['b', 'g', 'm','r', 'c', 'y', 'k']
+    for num, (key, value) in enumerate(seq_dict.items()):
+        color = colors_list[num % len(colors_list)]
+        plt.plot(x_windows[0], value, color = color, marker = "o", linestyle = "--", label = f"{key}")
+    plt.title("Sliding Window GC Content Calculation")
+    plt.xlabel("Sequence Windows")
+    plt.ylabel("GC content")
+    plt.legend()
+    plt.tight_layout()
+    plt.show()
+
+def create_window_plot(sequence):
     x_axis, y_axis = calculate_window_gc(sequence, window_size, step_size)
-    plt.plot(x_axis, y_axis, color = "k", linestyle = "--", marker = "o")
+    plt.plot(x_axis, y_axis, color = "k", linestyle = "--", marker = "o", label = f"{sequence}")
     plt.title("Sliding Window GC Content Calculation")
-    plt.xlabel("Sequence windows")
+    plt.xlabel("Sequence Windows")
     plt.ylabel("GC content")
+    plt.legend()
     plt.tight_layout()
     plt.show()
+
+def create_GC_barplot(x, y):
+    fig, ax = plt.subplots()
+    bar_container = ax.bar(x, y, width = 0.4, color = '#C04000')
+    ax.set_title("Sequence GC Content Analysis")
+    ax.set_xlabel("Sequences")
+    ax.set_ylabel("GC content")
+    ax.bar_label(bar_container)
+    ax.spines['top'].set_color('0.5')
+    ax.spines['right'].set_color('0.5')
+    ax.spines['bottom'].set_color('0.5')
+    ax.spines['left'].set_color('0.5')
+    fig.tight_layout()
+    plt.show()
 
+
 # ====================================================== LOGIC =====================================================================
-sequence = Seq(input('Please enter a DNA/RNA sequence. '))
-
-choice = input("Would you like to perform a sliding window GC content calculation? ")
-if choice.lower()[0] == 'n':
-    calculate_gc(sequence)
+print("Welcome to GC calculator!")
+seq_choice = int(input("Please enter the number of the sequences you want to analyze today. "))
+if seq_choice == 1:
+    sequence = Seq(input('Please enter the DNA/RNA sequence. '))
+    choice = input("Would you like to perform a sliding window GC content calculation? ")
+    if choice.lower()[0] == 'n':
+        calculate_gc(sequence)
+    else:
+        window_size = int(input("Please enter window size. "))
+        step_size = int(input("Please enter step size. "))
+        create_window_plot(sequence)
+
 else:
-    window_size = int(input("Please enter window size. "))
-    step_size = int(input("Please enter step size. "))
-    create_window_plot()
+    mult_choice = input("Would you like to perform a sliding window GC content calculation?")
+    if mult_choice.lower()[0] == 'n':
+        x_values , y_values = multiple_gc(seq_choice)
+        print(x_values)
+        print(y_values)
+        create_GC_barplot(x_values, y_values)
+
+    else:
+        print("Note! All sequences will be analyzed using the same window and step size.")
+        window_size = int(input("Please enter window size. "))
+        step_size = int(input("Please enter step size. "))
+        sequence_list = []
+        windows = []
+        gc_content_list = []
+        print("All sequences must be of the same length.")
+        for i in range(seq_choice):
+            new_seq = Seq(input("Please enter a DNA/RNA sequence. "))
+            sequence_list.append(new_seq)
+        for seq in sequence_list:
+            window , gc_content = calculate_window_gc(seq, window_size, step_size)
+            windows.append(window)
+            gc_content_list.append(gc_content)
+        # create dictionary from sequences and gc_values
+        pairs = zip(sequence_list, gc_content_list)
+        seq_dict = dict(pairs)
+        multi_window_plot(windows, seq_dict)
 
 
+