add pages

Dockerfile

@@ -30,4 +30,4 @@ EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-ENTRYPOINT ["streamlit", "run", "src/app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/Home.py", "--server.port=8501", "--server.address=0.0.0.0"]

Makefile

@@ -1,5 +1,5 @@
 run:
-	poetry run streamlit run src/app.py
+	poetry run streamlit run src/Home.py
 
 deps:
 	poetry export --without-hashes --without=dev --format=requirements.txt > requirements.txt

src/Home.py

@@ -0,0 +1,19 @@
+import streamlit as st
+
+
+# Page configuration
+st.set_page_config(
+    page_title="Wren AI Cloud API Demo",
+    page_icon="📊",
+    layout="wide",
+)
+
+st.title("📊 Wren AI Cloud API Demo")
+st.markdown(
+"""
+Please select a demo from the sidebar.
+
+- API Documentation: https://wrenai.readme.io/reference/cloud-getting-started
+- Demo Source Code: https://huggingface.co/spaces/getWrenAI/wrenai-cloud-api-demo/tree/main
+"""
+)

src/{app.py → pages/1_Chart_Generation.py}

@@ -3,17 +3,8 @@ import streamlit as st
 from apis import generate_sql, generate_chart
 
 
-# Page configuration
-st.set_page_config(
-    page_title="Wren AI Cloud API Demo",
-    page_icon="📊",
-    layout="wide"
-)
-
-
 def main():
-    st.title("📊 Wren AI Cloud API Demo")
-    st.markdown("Ask questions about your data and get both SQL queries and beautiful charts!")
+    st.title("📊 Wren AI Cloud API Demo - Chart Generation")
 
     # Sidebar for API configuration
     with st.sidebar:
@@ -169,4 +160,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main() 
+    main()