init

app.py

@@ -0,0 +1,52 @@
+import os
+import streamlit as st
+from PIL import Image
+
+from huggingface_hub import Repository
+from huggingface_hub import login
+
+login(token = os.environ['HF_TOKEN'])
+
+repo = Repository(
+    local_dir="agent_function",
+    repo_type="dataset",
+    clone_from=os.environ['DATASET'],
+    token=True
+)
+repo.git_pull()
+
+from agent_function.function import generate_answer
+
+Image.MAX_IMAGE_PIXELS = None
+
+st.set_page_config(layout="wide")
+
+st.title("Map2U Chatbot")
+col1, col2 = st.columns([1 ,2])
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+if "overlay" not in st.session_state:
+    st.session_state.overlay = 'base'
+
+with col1:
+    messages_box = st.container(height=500)
+        # Display chat messages from history on app rerun
+    for message in st.session_state.messages:
+        with messages_box.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    # React to user input
+    if prompt := st.chat_input("What is up?"):
+        messages_box.chat_message("user").markdown(prompt)
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        response = generate_answer(prompt)
+        with messages_box.chat_message("assistant"):
+            st.markdown(response)
+        # Add assistant response to chat history
+        st.session_state.messages.append({"role": "assistant", "content": response})
+
+    with col2:
+        image = Image.open(f'agent_function/{st.session_state.overlay}.png')
+        st.image(image)

requirements.txt

@@ -0,0 +1,68 @@
+affine==2.4.0
+altair==5.5.0
+annotated-types==0.7.0
+anyio==4.8.0
+attrs==24.3.0
+blinker==1.9.0
+cachetools==5.5.1
+certifi==2024.12.14
+charset-normalizer==3.4.1
+click==8.1.8
+click-plugins==1.1.1
+cligj==0.7.2
+contourpy==1.3.1
+cycler==0.12.1
+distro==1.9.0
+filelock==3.17.0
+fonttools==4.55.3
+fsspec==2024.12.0
+gitdb==4.0.12
+GitPython==3.1.44
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.27.1
+idna==3.10
+Jinja2==3.1.5
+jiter==0.8.2
+jsonschema==4.23.0
+jsonschema-specifications==2024.10.1
+kiwisolver==1.4.8
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+matplotlib==3.10.0
+mdurl==0.1.2
+narwhals==1.23.0
+numpy==2.2.1
+openai==1.59.9
+packaging==24.2
+pandas==2.2.3
+pillow==11.1.0
+protobuf==5.29.3
+pyarrow==19.0.0
+pydantic==2.10.5
+pydantic_core==2.27.2
+pydeck==0.9.1
+Pygments==2.19.1
+pyparsing==3.2.1
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+pytz==2024.2
+PyYAML==6.0.2
+rasterio==1.4.3
+referencing==0.36.1
+requests==2.32.3
+rich==13.9.4
+rpds-py==0.22.3
+shapely==2.0.6
+six==1.17.0
+smmap==5.0.2
+sniffio==1.3.1
+streamlit==1.41.1
+tenacity==9.0.0
+toml==0.10.2
+tornado==6.4.2
+tqdm==4.67.1
+typing_extensions==4.12.2
+tzdata==2025.1
+urllib3==2.3.0