Upload folder using huggingface_hub

src/connect4_backend.egg-info/PKG-INFO

@@ -0,0 +1,42 @@
+Metadata-Version: 2.1
+Name: connect4-backend
+Version: 0.0.1
+Author: Clement Gohlke
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: tensordict@ git+https://github.com/pytorch/tensordict.git@c3caa7612275306ce72697a82d5252681ddae0ab
+Requires-Dist: torchrl@ git+https://github.com/pytorch/rl.git@1bb192e0f3ad9e7b8c6fa769bfa3bb9d82ca4f29
+Requires-Dist: litrl~=0.0.9
+Requires-Dist: fastapi==0.104.1
+Requires-Dist: uvicorn==0.25.0
+Requires-Dist: moviepy==1.0.3
+Provides-Extra: test
+Requires-Dist: pytest==7.4.4; extra == "test"
+Requires-Dist: mypy==1.8.0; extra == "test"
+Requires-Dist: httpx==0.26.0; extra == "test"
+
+---
+title: Connect4
+emoji: 🌐
+colorFrom: blue
+colorTo: yellow
+sdk: docker
+pinned: false
+license: mit
+---
+
+frontend adapted from [github](https://github.com/jprioses/connect-four-game)
+
+Check out the configuration reference at [huggingface](https://huggingface.co/docs/hub/spaces-config-reference)
+
+## TODO
+
+link to github repo
+
+test block forced move
+
+Connect to AWS database https://aws.amazon.com/rds/?p=ft&c=db&z=3
+using alembic and sqlmodel

src/connect4_backend.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+pyproject.toml
+src/__init__.py
+src/app_state.py
+src/create_app.py
+src/typing.py
+src/connect4_backend.egg-info/PKG-INFO
+src/connect4_backend.egg-info/SOURCES.txt
+src/connect4_backend.egg-info/dependency_links.txt
+src/connect4_backend.egg-info/requires.txt
+src/connect4_backend.egg-info/top_level.txt

src/connect4_backend.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

src/connect4_backend.egg-info/requires.txt

@@ -0,0 +1,11 @@
+tensordict@ git+https://github.com/pytorch/tensordict.git@c3caa7612275306ce72697a82d5252681ddae0ab
+torchrl@ git+https://github.com/pytorch/rl.git@1bb192e0f3ad9e7b8c6fa769bfa3bb9d82ca4f29
+litrl~=0.0.9
+fastapi==0.104.1
+uvicorn==0.25.0
+moviepy==1.0.3
+
+[test]
+pytest==7.4.4
+mypy==1.8.0
+httpx==0.26.0

src/connect4_backend.egg-info/top_level.txt

@@ -0,0 +1,4 @@
+__init__
+app_state
+create_app
+typing

src/create_app.py

@@ -10,10 +10,17 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from litrl.env.connect_four import Board
 from loguru import logger
-from fastapi.responses import StreamingResponse, FileResponse
-from moviepy.editor import VideoFileClip
+from fastapi.responses import StreamingResponse
 from src.app_state import AppState
 from src.typing import CpuConfig
+from src.huggingface.huggingface_client import HuggingFaceClient
+
+def stream_mp4(mp4_path: Path) -> StreamingResponse:
+    def iter_file()-> Generator[bytes, Any, None]:  
+        with mp4_path.open(mode="rb") as env_file:
+            yield from env_file
+
+    return StreamingResponse(content=iter_file(), media_type="video/mp4")
 
 
 def create_app() -> FastAPI:
@@ -33,9 +40,8 @@ def create_app() -> FastAPI:
     def bot_action(
         env_id: SingleAgentId,
     ) -> str:
-        env = make(id=env_id, render_mode="rgb_array")
         env = RecordVideo(
-            env=env,
+            env=make(id=env_id, render_mode="rgb_array"),
             video_folder="tmp",
         )
         env.reset(seed=123)
@@ -46,14 +52,17 @@ def create_app() -> FastAPI:
             _, _, terminated, truncated, _ = env.step(action=action)
             env.render()
         env.video_recorder.close()
+        return stream_mp4(mp4_path=Path(env.video_recorder.path))
 
-        def iterfile()-> Generator[bytes, Any, None]:  
-            with Path(env.video_recorder.path).open(mode="rb") as env_file:
-                yield from env_file
-
-        return StreamingResponse(content=iterfile(), media_type="video/mp4")
+    @app.get(path=f"/hfmp4")
+    def fh_stream(
+        env_id: SingleAgentId,
+        hf_client: Annotated[HuggingFaceClient, Depends(dependency=HuggingFaceClient)],
+    ) -> StreamingResponse:
+        hf_client.mp4_paths[env_id]
+        return stream_mp4(mp4_path=hf_client.mp4_paths[env_id])
 
-    @app.get(path=f"/gif")
+    @app.get(path=f"/mp4")
     def bot_action(
         env_id: SingleAgentId,
     ) -> str:
@@ -70,12 +79,7 @@ def create_app() -> FastAPI:
             _, _, terminated, truncated, _ = env.step(action=action)
             env.render()
         env.video_recorder.close()
-
-        mp4_path = Path(env.video_recorder.path)
-        video_clip = VideoFileClip(str(mp4_path))
-        gif_path = mp4_path.with_suffix(".gif")
-        video_clip.write_gif(str(gif_path))#, fps=30)  # TODO check fps
-        return FileResponse(str(gif_path), media_type="image/gif")
+        return stream_mp4(mp4_path=Path(env.video_recorder.path))
 
     @app.exception_handler(exc_class_or_status_code=RequestValidationError)
     async def validation_exception_handler(

src/huggingface/huggingface_client.py

@@ -5,7 +5,7 @@ from .get_files import get_mp4_paths
 from .get_environments import get_environments
 
 
-class HugingFaceClient:
+class HuggingFaceClient:
     def __init__(self) -> None:
         login(  # type: ignore[no-untyped-call]
             token=os.environ.get("HUGGINGFACE_TOKEN"),