Fix the location of job submission button

app.py

@@ -5,12 +5,17 @@ from io import StringIO
 import joblib
 import numpy as np
 import pandas as pd
+
 # page set up
 import streamlit as st
 from b3clf.descriptor_padel import compute_descriptors
 from b3clf.geometry_opt import geometry_optimize
-from b3clf.utils import (get_descriptors, predict_permeability,
-                         scale_descriptors, select_descriptors)
+from b3clf.utils import (
+    get_descriptors,
+    predict_permeability,
+    scale_descriptors,
+    select_descriptors,
+)
 from streamlit_extras.let_it_rain import rain
 from streamlit_ketcher import st_ketcher
 
@@ -44,6 +49,9 @@ resample_methods_dict = {
     "no resampling": "common",
 }
 
+pandas_display_options = {
+    "line_limit": 50,
+}
 
 
 def generate_predictions(
@@ -147,8 +155,8 @@ with info_column:
                 data=file_smi,
                 file_name="sample_input_smiles.csv",
             )
-# Create a file uploader
 
+# Create a file uploader
 with upload_column:
     st.subheader("Model Selection")
     with st.container():
@@ -180,9 +188,9 @@ with upload_column:
         with upload_col:
             file = st.file_uploader(
                 label="Upload a CSV, SDF or TXT file",
-                type=["csv", "sdf", "txt"],
-                help="Input molecule file and only text files are supported.",
-                # accept_multiple_files=False,
+                type=["csv", "sdf", "txt", "smi"],
+                help="Input molecule file only supports *.csv, *.sdf, *.txt and *.smi.",
+                accept_multiple_files=False,
             )
         # submit job column
         with submit_job_col:
@@ -247,7 +255,10 @@ if submit_job_button:
 
         # feture table
         with feature_column:
-            st.dataframe(X_features)
+            selected_feature_rows = np.min(
+                [X_features.shape[0], pandas_display_options["line_limit"]]
+            )
+            st.dataframe(X_features.iloc[:selected_feature_rows, :], hide_index=False)
             # placeholder_features.dataframe(X_features, hide_index=False)
             feature_file_name = file.name.split(".")[0] + "_b3clf_features.csv"
             features_csv = X_features.to_csv(index=True)
@@ -262,7 +273,10 @@ if submit_job_button:
             # st.subheader("Predictions")
             if results is not None:
                 # Display the predictions in a table
-                st.dataframe(results, hide_index=True)
+                selected_result_rows = np.min(
+                    [results.shape[0], pandas_display_options["line_limit"]]
+                )
+                st.dataframe(results.iloc[:selected_result_rows, :], hide_index=True)
                 # Add a button to download the predictions as a CSV file
                 predictions_csv = results.to_csv(index=True)
                 results_file_name = file.name.split(".")[0] + "_b3clf_predictions.csv"