Spaces:
Runtime error
Runtime error
File size: 2,953 Bytes
105b369 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
from typing import Optional, Union, List, Dict
from phi.aws.app.base import AwsApp, ContainerContext, AwsBuildContext # noqa: F401
class Streamlit(AwsApp):
# -*- App Name
name: str = "streamlit"
# -*- Image Configuration
image_name: str = "phidata/streamlit"
image_tag: str = "1.27"
command: Optional[Union[str, List[str]]] = "streamlit hello"
# -*- App Ports
# Open a container port if open_port=True
open_port: bool = True
port_number: int = 8501
# -*- Workspace Configuration
# Path to the workspace directory inside the container
workspace_dir_container_path: str = "/usr/local/app"
# -*- ECS Configuration
ecs_task_cpu: str = "1024"
ecs_task_memory: str = "2048"
ecs_service_count: int = 1
assign_public_ip: Optional[bool] = True
# -*- Streamlit Configuration
# Server settings
# Defaults to the port_number
streamlit_server_port: Optional[int] = None
streamlit_server_headless: bool = True
streamlit_server_run_on_save: Optional[bool] = None
streamlit_server_max_upload_size: Optional[int] = None
streamlit_browser_gather_usage_stats: bool = False
# Browser settings
streamlit_browser_server_port: Optional[str] = None
streamlit_browser_server_address: Optional[str] = None
def get_container_env(self, container_context: ContainerContext, build_context: AwsBuildContext) -> Dict[str, str]:
container_env: Dict[str, str] = super().get_container_env(
container_context=container_context, build_context=build_context
)
streamlit_server_port = self.streamlit_server_port
if streamlit_server_port is None:
port_number = self.port_number
if port_number is not None:
streamlit_server_port = port_number
if streamlit_server_port is not None:
container_env["STREAMLIT_SERVER_PORT"] = str(streamlit_server_port)
if self.streamlit_server_headless is not None:
container_env["STREAMLIT_SERVER_HEADLESS"] = str(self.streamlit_server_headless)
if self.streamlit_server_run_on_save is not None:
container_env["STREAMLIT_SERVER_RUN_ON_SAVE"] = str(self.streamlit_server_run_on_save)
if self.streamlit_server_max_upload_size is not None:
container_env["STREAMLIT_SERVER_MAX_UPLOAD_SIZE"] = str(self.streamlit_server_max_upload_size)
if self.streamlit_browser_gather_usage_stats is not None:
container_env["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = str(self.streamlit_browser_gather_usage_stats)
if self.streamlit_browser_server_port is not None:
container_env["STREAMLIT_BROWSER_SERVER_PORT"] = self.streamlit_browser_server_port
if self.streamlit_browser_server_address is not None:
container_env["STREAMLIT_BROWSER_SERVER_ADDRESS"] = self.streamlit_browser_server_address
return container_env
|