feat: allow yaml input

examples/example/example.yaml

@@ -0,0 +1,6 @@
+- situation: "You don't know how to use this tool"
+  misery index: 80
+  image: umbrella.png
+- situation: "This is just an example"
+  misery index: 1
+  image: simple stick figure.png

requirements.txt

@@ -2,4 +2,5 @@ matplotlib
 pandas
 openpyxl
 tqdm
-streamlit
+streamlit
+pyyaml

src/app/main.py

@@ -73,6 +73,17 @@ def zip_cards(target_file: Path, source_dir: Path):
 
 with st.popover("Download example data"):
     st.write("Download example data to get started.")
+    with open(
+        Path(__file__).parent.parent.parent / "examples" / "example" / "example.yaml",
+        "rb",
+    ) as yaml_file:
+        st.download_button(
+            label="Input yaml",
+            data=yaml_file,
+            file_name="example.yaml",
+            mime="text/yaml",
+            icon=":material/download:",
+        )
     with open(
         Path(__file__).parent.parent.parent / "examples" / "example" / "example.csv",
         "rb",
@@ -139,9 +150,10 @@ def zip_cards(target_file: Path, source_dir: Path):
                 with open(image_dir / image.name, "wb") as image_file:
                     image_file.write(image.getbuffer())
 
-            expansion_logo_path = image_dir / expansion_logo.name
-            with open(expansion_logo_path, "wb") as logo_file:
-                logo_file.write(expansion_logo.getbuffer())
+            if expansion_logo is not None:
+                expansion_logo_path = image_dir / expansion_logo.name
+            else:
+                expansion_logo_path = None
 
             df = parse_input_file(input_file, image_dir)
 

src/ithappens/create_cards.py

@@ -17,6 +17,7 @@
 from matplotlib.transforms import Bbox
 from PIL import Image
 from tqdm import tqdm
+from yaml import safe_load
 
 
 from ithappens.card import Card
@@ -138,24 +139,44 @@ def parse_input_file(
         Pandas DataFrame with index, description, and misery index.
     """
     usecols = ["misery index", "situation", "image"]
+
+    df = None
     try:
-        df = pd.read_excel(input_path)
-        df = df[usecols]
+        with open(input_path, 'r') as f:
+            df = pd.json_normalize(safe_load(f))
+            df = df[usecols]
+    except TypeError:
+            bytes_data = input_path.getvalue()
+            df = pd.json_normalize(safe_load(bytes_data))
+            df = df[usecols]
     except ValueError:
         pass
     except KeyError:
-        print(f"Make sure {input_path} has two columns named {usecols}.")
+        print(f"yaml: Make sure {input_path} has two columns named {usecols}.")
         exit()
+    except Exception as e:
+        raise e
 
-    try:
-        df = pd.read_csv(input_path)
-        df = df[usecols]
-    except UnicodeDecodeError:
-        print(f"{input_path} is not a valid .csv or .xlsx file.")
-        exit()
-    except KeyError:
-        print(f"Make sure {input_path} has two columns named {usecols}.")
-        exit()
+    if df is None:
+        try:
+            df = pd.read_excel(input_path)
+            df = df[usecols]
+        except ValueError:
+            pass
+        except KeyError:
+            print(f"excel: Make sure {input_path} has two columns named {usecols}.")
+            exit()
+
+    if df is None:
+        try:
+            df = pd.read_csv(input_path)
+            df = df[usecols]
+        except UnicodeDecodeError:
+            print(f"{input_path} is not a valid .csv or .xlsx file.")
+            exit()
+        except KeyError:
+            print(f"csv: Make sure {input_path} has two columns named {usecols}.")
+            exit()
 
     if image_dir: