implement frontend app

Author: ks6088ts

frontend/entrypoint.py

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

frontend/solutions/azure_ai_vision.py

@@ -0,0 +1,45 @@
+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 AI Vision")
+
+    file_uploader = st.file_uploader(
+        label="Choose a file",
+        key="file_uploader",
+    )
+
+    analyze_button = st.button(
+        label="Analyze",
+        key="analyze_button",
+    )
+
+    if file_uploader is not None:
+        st.image(file_uploader, caption="Uploaded image")
+        if analyze_button:
+            with st.spinner("Analyzing..."):
+                bytes_data = file_uploader.getvalue()
+                response = asyncio.run(
+                    http_post_file(
+                        url=urljoin(base=backend_url, url="/azure_ai_vision/image/analyze/"),
+                        data_bytes_io=BytesIO(bytes_data),
+                    )
+                )
+                st.write(response)
+    else:
+        st.warning("Please upload a file first")

frontend/solutions/types.py

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