Update app.py

app.py

@@ -59,6 +59,7 @@ def find_new_boundary_repeats(fragments, final_repeats, overlap=50):
 def process_protein_sequence(sequence, analysis_type, overlap=50):
     fragments = fragment_protein_sequence(sequence)
     final_repeats = defaultdict(int)
+    homo_repeats = {}
     for fragment in fragments:
         if analysis_type in ["Hetero", "Both"]:
             fragment_repeats = find_hetero_amino_acid_repeats(fragment)
@@ -70,9 +71,15 @@ def process_protein_sequence(sequence, analysis_type, overlap=50):
         for k, v in new_repeats.items():
             final_repeats[k] += v
     if analysis_type in ["Homo", "Both"]:
-        homorepeats = find_homorepeats(sequence)
-        for k, v in homorepeats.items():
+        homo_repeats = find_homorepeats(sequence)
+        for k, v in homo_repeats.items():
             final_repeats[k] += v
+    if analysis_type == "Hetero" or "Both":
+        for k, v in homo_repeats.items():
+            if k in final_repeats:
+                final_repeats[k] -= v
+                if final_repeats[k] <= 0:
+                    del final_repeats[k]
     return final_repeats
 
 def process_excel(excel_data, analysis_type):