Minor tweaks

app.py

@@ -28,7 +28,7 @@ def main():
         image_file_hand = st.file_uploader("Your hand")
         # image_file_hand = image_file = st.camera_input("Your hand")
         if image_file_hand is not None:
-            preds = _run_inference(image_file_hand)
+            preds = inference(image_file_hand)
             if len(preds) != 2:
                 _show_error_message(2, preds, image_file_hand, "hand")
                 image_file_hand = None
@@ -38,7 +38,7 @@ def main():
         image_file_flop = st.file_uploader("Flop")
         # image_file_flop = image_file = st.camera_input(label="Flop")
         if image_file_flop is not None:
-            preds = _run_inference(image_file_flop)
+            preds = inference(image_file_flop)
             if len(preds) != 3:
                 _show_error_message(3, preds, image_file_flop, "flop")
                 image_file_flop = None
@@ -65,8 +65,7 @@ def main():
     if not _validate_cards(hand, flop):
         return
     run_simulation(hand=hand, flop=flop)
-    st.divider()
-    st.write("Interested in building more Computer Vision applications? Check out the [LandingLens](https://landing.ai/) platform and our [open source Python SDK](https://github.com/landing-ai/landingai-python)!")
+    st.write("Interested in building more Computer Vision applications? Check out our [LandingLens](https://landing.ai/) platform and our [open source Python SDK](https://github.com/landing-ai/landingai-python)!")
 
 
 def _validate_cards(hand, flop) -> bool:
@@ -93,18 +92,14 @@ def _convert_name(name: str) -> str:
 
 
 # TODO Rename this here and in `main`
-def _show_error_message(expected_length, preds, img_file, pred_name):
-    msg = (
-        f"Missing {expected_length - len(preds)} card in your {pred_name}. Please try again with a new photo."
-        if len(preds) < expected_length
-        else f"Detected more than {len(preds) - expected_length} card in your {pred_name}. Please try again with a new photo."
-    )
+def _show_error_message(expected_len, preds, img_file, pred_name):
+    msg = f"Detected {len(preds)}, expects {expected_len} cards in your {pred_name}. Please try again with a new photo."
     st.error(msg)
     _show_predictions(preds, img_file, pred_name)
 
 
 @st.cache_data
-def _run_inference(image_file) -> List[Prediction]:
+def inference(image_file) -> List[Prediction]:
     image = Image.open(image_file).convert("RGB")
     predictor = Predictor(endpoint_id="ecab49f5-9047-4d5a-8c92-b7ad8e908324")
     logging.info("Running Poker prediction")

holdem.py

@@ -1,7 +1,8 @@
+import streamlit as st
 from prettytable import PrettyTable
-from poker_functions import Card
+
 import simulation as s
-import streamlit as st
+from poker_functions import Card
 
 
 def run_simulation(
@@ -9,9 +10,9 @@ def run_simulation(
     flop: list[str] = None,
     turn: list[str] = None,
     river: list[str] = None,
-    sims=100000,
+    sims=10000,
 ) -> None:
-    sim = s.simulation_one_player(hand, flop, turn, river, sims)
+    sim = s.simulation_one_player(hand, flop, turn, river, sims=sims)
     hc_pct = s.percent(sim[1], sim[0])
     hc_ratio = s.ratio(sim[1], sim[0])
     pair_pct = s.percent(sim[2], sim[0])
@@ -94,5 +95,6 @@ def run_simulation(
     st.write(odds, "\n")
     st.divider()
 
+
 if __name__ == "__main__":
-    run_simulation(["As", "Ah"], ["2s", "3s", "4s"], sims=10000)
+    run_simulation(["As", "Ah"], ["2s", "3s", "4s"], sims=4000)