0.7.5 (#27)

* GraphGPT input field replaced with text area. Better JSON parsing added

* Additional UI data output

* Dependencies updated

* Sample images added. 

* Generator Radio toggles fix

README.md

@@ -15,13 +15,5 @@ poetry update
 poetry run streamlit run graph_data_generator_streamlit/app.py
 ```
 
-or 
-
-```
-pipenv shell
-pipenv sync
-pipenv run streamlit run graph_data_generator_streamlit/app.py 
-```
-
 ## Testing with local packages
 `poetry add --editable /path/to/package`

graph_data_generator_streamlit/app.py

@@ -5,13 +5,18 @@
 from ui.design_ui import arrows_ui, generators_ui
 from ui.ideate_ui import ideate_ui
 from ui.export_ui import export_ui
+from ui.samples_ui import samples_list
+from graph_data_generator import start_logging
 import logging
 
 # SETUP
 st.set_page_config(layout="wide",initial_sidebar_state='collapsed')
 logging.getLogger().setLevel(logging.DEBUG)
 logging.info(f'App Started')
 
+# Uncomment to start graph_data_generator logging
+# start_logging()
+
 # LOAD any env
 neo4j_uri = st.secrets.get("NEO4J_URI", None)
 if "NEO4J_URI" not in st.session_state:
@@ -37,6 +42,8 @@
     st.session_state["ARROWS_DICT"] = None
 if "JSON_CONFIG" not in st.session_state:
     st.session_state["JSON_CONFIG"] = None
+if "SAMPLE_IMAGES" not in st.session_state:
+    st.session_state["SAMPLE_IMAGES"] = []
 
 
 # Header
@@ -69,4 +76,8 @@
 
 # Side bar
 with st.sidebar:
-    generators_ui()
+    tab1, tab2 = st.tabs(["Generators", "Samples"])
+    with tab1:
+        generators_ui()
+    with tab2:
+        samples_list()

graph_data_generator_streamlit/ui/design_ui.py

@@ -72,9 +72,13 @@ def generators_ui():
                         key = f'{generator.name}_{arg.label}'
                         ))
                 if arg.type == GeneratorType.BOOL:
+                    options = ["True", "False"]
+                    default = str(arg.default)
+                    default_index = options.index(default)
                     arg_inputs.append(st.radio(
                         label=arg.label,
-                        index=arg.default,
+                        index=default_index,
+                        options = options,
                         key = f'{generator.name}_{arg.label}'
                     ))
                 if arg.type == GeneratorType.DATETIME:

graph_data_generator_streamlit/ui/samples_ui.py

@@ -0,0 +1,22 @@
+import streamlit as st
+import os
+import logging
+from PIL import Image
+
+directory = 'graph_data_generator_streamlit/images/'
+
+
+def samples_list():
+    if len(st.session_state["SAMPLE_IMAGES"]) == 0:
+        filenames = os.listdir(directory)
+        filenames.sort()
+        for filename in filenames:
+            try:
+                image = Image.open(os.path.join(directory, filename))
+                st.image(image, caption=f'{filename}')
+                st.session_state["SAMPLE_IMAGES"].append((image, filename))
+            except Exception as e:
+                logging.error(e)
+    else:
+        for image_filename in st.session_state["SAMPLE_IMAGES"]:
+            st.image(image_filename[0], caption=image_filename[1])

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mock-graph-data-generator-streamlit"
-version = "0.7.3"
+version = "0.7.5"
 description = ""
 authors = ["Jason Koo <[email protected]>"]
 readme = "README.md"
@@ -9,17 +9,19 @@ packages = [{include = "graph_data_generator_streamlit"}]
 [tool.poetry.dependencies]
 python = "^3.11"
 streamlit = ">=1.27,<2.0"
-neo4j-uploader = {path = "/Users/jasonkoo/neo4j/repos/neo4j-uploader", develop = true}
 pyclip = ">=0.7,<1.0"
-graph-data-generator = ">=0.3.2"
 neo4j = "^5.14.1"
 streamlit-agraph = "^0.0.45"
 openai = "^0.28.1"
+opencv-python = "^4.8.1.78"
+pillow = "^10.1.0"
+graph-data-generator = "0.4.0"
+neo4j-uploader = "^0.4.1"
 
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.4.0"
 
 [build-system]
 requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
+build-backend = "poetry.core.masonry.api"