Update app.py

app.py

@@ -66,27 +66,44 @@ if uploaded_file is not None:
     if st.button(f'Design Primers for {feature_type}'):
         size_range = [tuple(map(int, r.split('-'))) for r in product_size_range.split(',')]
         primers = design_primers_for_region(record.seq, start, end, num_to_return=5)
-        primer_df = pd.DataFrame({
-            'Primer': ['Left Primer'] * 5 + ['Right Primer'] * 5,
-            'Sequence': [primers[f'PRIMER_LEFT_{i}_SEQUENCE'] for i in range(5)] + 
-                        [primers[f'PRIMER_RIGHT_{i}_SEQUENCE'] for i in range(5)],
-            'Tm (°C)': [primers[f'PRIMER_LEFT_{i}_TM'] for i in range(5)] + 
-                       [primers[f'PRIMER_RIGHT_{i}_TM'] for i in range(5)],
-            'Length': [primers[f'PRIMER_LEFT_{i}_SIZE'] for i in range(5)] +
-                      [primers[f'PRIMER_RIGHT_{i}_SIZE'] for i in range(5)],
-            'GC%': [primers[f'PRIMER_LEFT_{i}_GC_PERCENT'] for i in range(5)] +
-                   [primers[f'PRIMER_RIGHT_{i}_GC_PERCENT'] for i in range(5)],
-        })
+        
+        # Initialize lists to hold primer details
+        sequences, tm_values, lengths, gc_percents = [], [], [], []
+        
+        # Check if primer keys exist and add their details to the lists
+        for i in range(5):  # Assuming we are looking for 5 primer pairs
+            left_key = f'PRIMER_LEFT_{i}_SEQUENCE'
+            right_key = f'PRIMER_RIGHT_{i}_SEQUENCE'
+            if left_key in primers and right_key in primers:
+                sequences.append(primers[left_key])
+                sequences.append(primers[right_key])
+                tm_values.append(primers[f'PRIMER_LEFT_{i}_TM'])
+                tm_values.append(primers[f'PRIMER_RIGHT_{i}_TM'])
+                lengths.append(primers[f'PRIMER_LEFT_{i}_SIZE'])
+                lengths.append(primers[f'PRIMER_RIGHT_{i}_SIZE'])
+                gc_percents.append(primers[f'PRIMER_LEFT_{i}_GC_PERCENT'])
+                gc_percents.append(primers[f'PRIMER_RIGHT_{i}_GC_PERCENT'])
+        
+        # Check if any primers were found
+        if sequences:
+            # Construct the DataFrame using the collected primer details
+            primer_df = pd.DataFrame({
+                'Primer': ['Left Primer'] * len(sequences) // 2 + ['Right Primer'] * len(sequences) // 2,
+                'Sequence': sequences,
+                'Tm (°C)': tm_values,
+                'Length': lengths,
+                'GC%': gc_percents,
+            })
 
-        st.write('### Designed Primers')
-        st.dataframe(primer_df)
-        csv = primer_df.to_csv(index=False).encode('utf-8')
-        st.download_button(
-            "Download Primers as CSV",
-            csv,
-            "primers.csv",
-            "text/csv",
-            key='download-csv'
-        )
-else:
-    st.warning("Please upload a GenBank file.")
+            st.write('### Designed Primers')
+            st.dataframe(primer_df)
+            csv = primer_df.to_csv(index=False).encode('utf-8')
+            st.download_button(
+                "Download Primers as CSV",
+                csv,
+                "primers.csv",
+                "text/csv",
+                key='download-csv'
+            )
+        else:
+            st.error('No primers were found. Please adjust your parameters and try again.')