More polish and change camera input widget

app.py

@@ -1,7 +1,6 @@
 import logging
 from typing import List
 
-import numpy as np
 import streamlit as st
 from landingai.common import Prediction
 from landingai.predict import Predictor
@@ -9,8 +8,8 @@ from landingai.st_utils import render_svg, setup_page
 from landingai.visualize import overlay_predictions
 from PIL import Image
 
-from holdem import run_simulation
 import simulation as s
+from holdem import run_simulation
 
 setup_page(page_title="LandingLens Holdem Odds Calculator")
 
@@ -26,8 +25,8 @@ def main():
     # image_file = st.camera_input("Camera View")
     tab1, tab2 = st.tabs(["Your hand", "Flop"])
     with tab1:
-        # image_file_hand = st.file_uploader("Your hand")
-        image_file_hand = image_file = st.camera_input("Your hand")
+        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)
             if len(preds) != 2:
@@ -36,8 +35,8 @@ def main():
                 return
             st.session_state[_HAND] = preds, image_file_hand
     with tab2:
-        # image_file_flop = st.file_uploader("Flop")
-        image_file_flop = image_file = st.camera_input(label="Flop")
+        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)
             if len(preds) != 3:
@@ -66,15 +65,22 @@ 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)!")
 
 
 def _validate_cards(hand, flop) -> bool:
     check = hand + flop
     if s.dedup(check):
-        st.error("There is a duplicate card.  Please check the board and your hand and try again.", icon="🚨")
+        st.error(
+            "There is a duplicate card.  Please check the board and your hand and try again.",
+            icon="🚨",
+        )
         return False
     if not s.validate_card(check):
-        st.error("At least one of your cards is not valid.  Please try again.", icon="🚨")
+        st.error(
+            "At least one of your cards is not valid.  Please try again.", icon="🚨"
+        )
         return False
     return True
 
@@ -85,6 +91,7 @@ def _convert_name(name: str) -> str:
     else:
         return f"{name[0].upper()}{name[1:].lower()}"
 
+
 # TODO Rename this here and in `main`
 def _show_error_message(expected_length, preds, img_file, pred_name):
     msg = (

holdem.py

@@ -1,4 +1,5 @@
 from prettytable import PrettyTable
+from poker_functions import Card
 import simulation as s
 import streamlit as st
 
@@ -30,7 +31,7 @@ def run_simulation(
     strt_flush_pct = s.percent(sim[9], sim[0])
     strt_flush_ratio = s.ratio(sim[9], sim[0])
 
-    hole_card_str = f"{hand[0]} {hand[1]}"
+    hole_card_str = f"{Card(hand[0]).pretty_name} {Card(hand[1]).pretty_name}"
 
     table = PrettyTable()
     table.field_names = ["Your Hand", "Board"]
@@ -40,7 +41,7 @@ def run_simulation(
         turn = []
     if river is None:
         river = []
-    board_str = "".join(f"{card} " for card in (flop + turn + river))
+    board_str = "".join(f"{Card(card).pretty_name} " for card in (flop + turn + river))
     table.add_row([hole_card_str, board_str])
 
     odds = PrettyTable()

poker_functions.py

@@ -33,6 +33,17 @@ class Card:
     def __str__(self):
         return self.name
 
+    @property
+    def pretty_name(self):
+        rank = '10' if self.rank == 'T' else self.rank
+        suit_map = {
+            "c": "♣",
+            "d": "♦",
+            "h": "♥",
+            "s": "♠",
+        }
+        return f'{rank}{suit_map[self.suit]}'
+
 
     def __getitem__(self, item):
         if item == 'rank':