Update app.py

app.py

@@ -188,7 +188,6 @@ class Stats:
                 if uid[4:] not in ids:
                     result = self.data[uid].copy()
                     result["link"] = f'https://www.baseball-reference.com/players/{result["nameLast"].lower()[0]}/{result["bbrefID"]}.shtml'
-                    result["yearID"] = str(result["yearID"])
                     results.append(result)
                     ids.add(uid[4:])
 
@@ -445,7 +444,15 @@ class Application:
         """
         )
 
-        self.player()
+        player, search = st.tabs(["Player", "Search"])
+
+        # Player tab
+        with player:
+            self.player()
+
+        # Search
+        with search:
+            self.search()
 
     def player(self):
         """
@@ -478,11 +485,36 @@ class Application:
         results = stats.search(name, year)
 
         # Display results
-        self.table(results, ["nameFirst", "nameLast", "teamID"] + stats.columns[1:] + ["link"])
-        
+        self.table(results, ["link", "nameFirst", "nameLast", "teamID"] + stats.columns[1:])
+
         # Save parameters
         st.experimental_set_query_params(category=category, name=name, year=year)
 
+    def search(self):
+        """
+        Stats search tab.
+        """
+
+        st.markdown("Find players with similar statistics.")
+
+        category = self.category("Batting", "searchcategory")
+        with st.form("search"):
+            if category == "Batting":
+                stats, columns = self.batting, self.batting.columns[:-6]
+            elif category == "Pitching":
+                stats, columns = self.pitching, self.pitching.columns[:-2]
+
+            # Enter stats with data editor
+            inputs = st.data_editor(pd.DataFrame([dict((column, None) for column in columns)]), hide_index=True).astype(float)
+
+            submitted = st.form_submit_button("Search")
+            if submitted:
+                # Run search
+                results = stats.search(row=inputs.to_dict(orient="records")[0])
+
+                # Display table
+                self.table(results, ["link", "nameFirst", "nameLast", "teamID"] + stats.columns[1:])
+
     def params(self):
         """
         Get application parameters. This method combines URL parameters with session parameters.
@@ -607,7 +639,20 @@ class Application:
         """
 
         if results:
-            st.dataframe(pd.DataFrame(results)[columns])
+            st.dataframe(
+                results,
+                column_order=columns,
+                column_config={
+                    "link": st.column_config.LinkColumn("Link", width="small"),
+                    "yearID": st.column_config.NumberColumn("Year", format="%d"),
+                    "nameFirst": "First",
+                    "nameLast": "Last",
+                    "teamID": "Team",
+                    "age": "Age",
+                    "weight": "Weight",
+                    "height": "Height",
+                },
+            )
         else:
             st.write("Player-Year not found")