Add initial project files including README, workflows, app, and requirements

.github/workflows/check-size.yml

@@ -0,0 +1,16 @@
+name: Check file size
+on:               # or directly `on: [push]` to run the action on every push on any branch
+  pull_request:
+    branches: [new]
+
+  # to run this workflow manually from the Actions tab
+  workflow_dispatch:
+
+jobs:
+  sync-to-hub:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check large files
+        uses: ActionsDesk/[email protected]
+        with:
+          filesizelimit: 10485760 # this is 10MB so we can sync to HF Spaces

.github/workflows/main.yml

@@ -0,0 +1,20 @@
+name: Sync to Hugging Face hub
+on:
+  push:
+    branches: [main]
+
+  # to run this workflow manually from the Actions tab
+  workflow_dispatch:
+
+jobs:
+  sync-to-hub:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+          # lfs: true
+      - name: Push to hub
+        env:
+          HF_TOKEN: ${{ secrets.HF_TOKEN }}
+        run: git push https://OxbridegeEcon:[email protected]/spaces/Oxbridge-Economics/watchpit-st main --force

README.md

@@ -1 +1,31 @@
-# watchpit-st
+---
+title: watchpit - prototype
+emoji: 🔥
+colorFrom: green
+colorTo: pink
+sdk: streamlit
+sdk_version: 1.42.0 # The latest supported version
+app_file: app.py
+pinned: false
+---
+Check out the configuration reference at <https://huggingface.co/docs/hub/spaces-config-reference>
+
+# 🎈 Blank app template
+
+A simple Streamlit app template for you to modify!
+
+[![Open in Streamlit](https://static.streamlit.io/badges/streamlit_badge_black_white.svg)](https://blank-app-template.streamlit.app/)
+
+### How to run it on your own machine
+
+1. Install the requirements
+
+   ```
+   $ pip install -r requirements.txt
+   ```
+
+2. Run the app
+
+   ```
+   $ streamlit run app.py
+   ```

app.py

@@ -0,0 +1,100 @@
+"""Streamlit app example."""
+import os
+import uuid
+import json
+import requests
+import streamlit as st
+
+DOMAIN = "https://oxbridge-economics-watchpit.hf.space"
+HEADERS = {
+    'Content-Type': 'application/json',
+    'Authorization': f'Bearer {os.environ.get("HF_TOKEN")}'
+}
+# STOCK = ["AAPL","ADBE","AMD","AMZN","ARM","ASML","AVGO","BRK.B","BTC","CHAU","COIN","CVNA",
+#          "DXYZ","GBTC","GOOG","GOOGL","KULR","MARA","META","MRVL","MSFT","MSTR","NBIS",
+#          "NFLX","NVDA","PDD","PLTR","QCOM","QQQ","QUBT","RDDT","ROKU","RR","SPY","TEM","TSLA"]
+
+if 'chat_id' not in st.session_state:
+    st.session_state.chat_id = str(uuid.uuid4())
+    st.session_state.user_id = str(uuid.uuid4())
+
+# with st.sidebar:
+#     st.header("Symbols Selection")
+#     symbols = st.multiselect("Select stocks", STOCK, default=["AAPL"])
+#     mode = st.selectbox("Mode", ["research", "stream"])
+#     retrivers = st.multiselect("Retrievers", ["duckduckgo", "google"], default=["duckduckgo"])
+
+# def get_research(query):
+#     """
+#     Sends a POST request to the research endpoint with the given query.
+
+#     Args:
+#         query (str): The search query to be sent in the request payload.
+
+#     Returns:
+#         str: The response text from the research endpoint.
+
+#     Raises:
+#         requests.exceptions.RequestException: If there is an issue with the request.
+#     """
+#     payload = json.dumps({
+#         "query": query,
+#         "symbols": symbols,
+#         "headers": {
+#             "retrievers": ",".join(retrivers)
+#         }
+#     })
+#     response = requests.request(
+#         "POST", f"{DOMAIN}/research", headers=HEADERS, data=payload, timeout=100)
+#     response = response.json()
+#     print(response["report"])
+#     return response["report"]
+
+def get_answer(query):
+    """
+    Sends a query to a specified URL and retrieves the answer from the response.
+
+    Args:
+        query (str): The query string to be sent.
+        symbols (list): The symbols to be sent
+
+    Returns:
+        str: The answer extracted from the response.
+    """
+    payload = json.dumps({
+            "query": query,
+            "chat_id": st.session_state.chat_id,
+            "user_id": st.session_state.user_id,
+            })
+    contexts = []
+    response_answer = ""
+    api_response = requests.request(
+        "POST", f"{DOMAIN}/stream", headers=HEADERS,data=payload, stream=True, timeout=60)
+    for response_str in api_response.text.strip().split('\n\n'):
+        response_list = response_str.split('\n')
+        response_str = response_list[1]
+        json_object = json.loads(response_str.replace("data: ", ""))
+        if "answer" in json_object:
+            response_answer = response_answer + json_object['answer']
+        elif "context" in json_object:
+            contexts.append(json_object['context'])
+        # elif "questions" in json_object:
+        #     questions = json_object['questions']
+    return response_answer
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+if prompt := st.chat_input("What is up?"):
+
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+    answer = get_answer(prompt)
+    with st.chat_message("assistant"):
+        response_content = st.markdown(answer)
+    st.session_state.messages.append({"role": "assistant", "content": answer})

requirements.txt

@@ -0,0 +1,2 @@
+streamlit
+requests