Update app.py

app.py

@@ -1004,11 +1004,6 @@ def unwrap_text(text):
     return text.strip.replece('\n', '')
 
 
-def smiles_from_sdf(sdf_path):
-    with Chem.SDMolSupplier(sdf_path) as suppl:
-        return Chem.MolToSmiles(suppl[0])
-
-
 def drug_library_from_sdf(sdf_path):
     return PandasTools.LoadSDF(
         sdf_path,
@@ -1528,8 +1523,15 @@ with gr.Blocks(theme=theme, title='DeepSEQreen', css=CSS) as demo:
         finally:
             return fasta_seq
 
+    def process_fasta_upload(fasta_upload):
+        fasta = ''
+        try:
+            fasta = fasta_upload.decode()
+        except Exception as e:
+            gr.Warning(f"Please upload a valid FASTA file. Error: {str(e)}")
+            return fasta
 
-    target_upload_btn.upload(fn=lambda x: x.decode(), inputs=target_upload_btn, outputs=target_fasta)
+    target_upload_btn.upload(fn=process_fasta_upload, inputs=target_upload_btn, outputs=target_fasta)
     target_query_btn.click(uniprot_query,
                            inputs=[target_input_type, target_id, target_gene, target_organism],
                            outputs=target_fasta)
@@ -1633,12 +1635,17 @@ QALAHAYFAQYHDPDDEPVADPYDQSFESRDLLIDEWKSLTYDEVISFVPPPLDQEEMES
 
 
     def compound_upload_process(input_type, input_upload):
-        match input_type:
-            case 'SMILES':
-                return input_upload.decode()
-            case 'SDF':
-                suppl = Chem.ForwardSDMolSupplier(io.BytesIO(input_upload))
-                return Chem.MolToSmiles(next(suppl))
+        smiles = ''
+        try:
+            match input_type:
+                case 'SMILES':
+                    smiles = input_upload.decode()
+                case 'SDF':
+                    suppl = Chem.ForwardSDMolSupplier(io.BytesIO(input_upload))
+                    smiles = Chem.MolToSmiles(next(suppl))
+        except Exception as e:
+            gr.Warning(f"Please upload a valid {input_type} file. Error: {str(e)}")
+        return smiles
 
 
     compound_upload_btn.upload(fn=compound_upload_process,