add solution to frontend

Author: ks6088ts

frontend/entrypoint.py

@@ -1,6 +1,6 @@
 import logging
 
-from frontend.solutions import document_intelligence, sandbox, transcription
+from frontend.solutions import azure_storage, document_intelligence, sandbox, transcription
 from frontend.solutions.types import SolutionType
 
 logger = logging.getLogger(__name__)
@@ -26,3 +26,8 @@ def start(
             backend_url=backend_url,
             log_level=log_level,
         )
+    if solution_type == SolutionType.AZURE_STORAGE:
+        return azure_storage.start(
+            backend_url=backend_url,
+            log_level=log_level,
+        )

frontend/solutions/azure_storage.py

@@ -0,0 +1,47 @@
+import asyncio
+import logging
+from io import BytesIO
+from urllib.parse import urljoin
+
+import streamlit as st
+
+from frontend.solutions.utilities import http_post_file
+
+logger = logging.getLogger(__name__)
+
+
+def start(
+    backend_url: str,
+    log_level: int,
+):
+    logger.setLevel(log_level)
+    logger.debug(f"set log level to {log_level}")
+
+    st.header("Azure Storage")
+
+    file_uploader = st.file_uploader(
+        label="Choose a file",
+        key="file_uploader",
+    )
+    blob_name = st.text_input(
+        label="Blob Name",
+        key="blob_name",
+    )
+    upload_button = st.button(
+        label="Upload File",
+        key="upload_button",
+    )
+
+    if upload_button:
+        if file_uploader is None:
+            st.warning("Please upload a file first")
+        else:
+            with st.spinner("Uploading..."):
+                bytes_data = file_uploader.getvalue()
+                response = asyncio.run(
+                    http_post_file(
+                        url=urljoin(base=backend_url, url=f"/azure_storage/blobs/upload/?blob_name={blob_name}"),
+                        data_bytes_io=BytesIO(bytes_data),
+                    )
+                )
+                st.write(response)

frontend/solutions/types.py

@@ -5,3 +5,4 @@ class SolutionType(Enum):
     SANDBOX = "SANDBOX"
     TRANSCRIPTION = "TRANSCRIPTION"
     DOCUMENT_INTELLIGENCE = "DOCUMENT_INTELLIGENCE"
+    AZURE_STORAGE = "AZURE_STORAGE"