Better error message

app.py

@@ -35,6 +35,12 @@ def process_inputs(smiles: str, file_obj):
             smiles_col = smiles_cols[0]
             smiles_list = df_in[smiles_col].astype(str).tolist()
 
+            if (len(smiles_list) > 100):
+                return (
+                    "Error: The CSV must have up to 100 Smiles.",
+                    gr.update(visible=False),
+                )
+
             embeddings = []
             for sm in smiles_list:
                 vec = model.encode(sm, return_torch=True)[0].tolist()
@@ -62,8 +68,7 @@ def process_inputs(smiles: str, file_obj):
         df_out.to_csv("embeddings.csv", index=False)
         return json.dumps(vec), gr.update(value="embeddings.csv", visible=True)
     except Exception as e:
-        return f"Error extracting embedding: {e}", gr.update(visible=False)
-
+        return f"The following input '{smiles}' is not a valid SMILES"
 
 # 4) Build the Gradio Blocks interface
 with gr.Blocks() as demo: