Upload folder using huggingface_hub

src/chattr/__init__.py

@@ -17,7 +17,9 @@ MCP_VOICE_GENERATOR: str = getenv(
 MCP_VIDEO_GENERATOR: str = getenv(
     key="MCP_VIDEO_GENERATOR", default="http://localhost:8002/"
 )
-VECTOR_DATABASE_NAME: str = getenv(key="VECTOR_DATABASE_NAME", default="chattr")
+VECTOR_DATABASE_NAME: str = getenv(
+    key="VECTOR_DATABASE_NAME", default="chattr"
+)
 DOCKER_MODEL_RUNNER_URL: str = getenv(
     key="DOCKER_MODEL_RUNNER_URL", default="http://127.0.0.1:12434/engines/v1"
 )
@@ -25,7 +27,9 @@ DOCKER_MODEL_RUNNER_MODEL_NAME: str = getenv(
     key="DOCKER_MODEL_RUNNER_MODEL_NAME",
     default="ai/qwen3:0.6B-Q4_0",
 )
-GROQ_URL: str = getenv(key="MODEL_URL", default="https://api.groq.com/openai/v1")
+GROQ_URL: str = getenv(
+    key="MODEL_URL", default="https://api.groq.com/openai/v1"
+)
 GROQ_MODEL_NAME: str = getenv(key="GROQ_MODEL_NAME", default="llama3-70b-8192")
 
 BASE_DIR: Path = Path.cwd()
@@ -56,7 +60,9 @@ MODEL_NAME: str = (
     else GROQ_MODEL_NAME
 )
 MODEL_API_KEY: str = (
-    "not-needed" if MODEL_URL == DOCKER_MODEL_RUNNER_URL else getenv("GROQ_API_KEY")
+    "not-needed"
+    if MODEL_URL == DOCKER_MODEL_RUNNER_URL
+    else getenv("GROQ_API_KEY")
 )
 MODEL_TEMPERATURE: float = float(getenv(key="MODEL_TEMPERATURE", default=0.0))
 

src/chattr/graph.py

@@ -46,7 +46,9 @@ async def create_graph() -> CompiledStateGraph:
         )
         _model = _model.bind_tools(_tools, parallel_tool_calls=False)
     except Exception as e:
-        raise RuntimeError(f"Failed to initialize ChatOpenAI model: {e}") from e
+        raise RuntimeError(
+            f"Failed to initialize ChatOpenAI model: {e}"
+        ) from e
 
     def call_model(state: MessagesState) -> MessagesState:
         """
@@ -58,7 +60,9 @@ async def create_graph() -> CompiledStateGraph:
         Returns:
             MessagesState: A new state with the model's response appended to the messages.
         """
-        return {"messages": [_model.invoke([SYSTEM_MESSAGE] + state["messages"])]}
+        return {
+            "messages": [_model.invoke([SYSTEM_MESSAGE] + state["messages"])]
+        }
 
     _builder: StateGraph = StateGraph(MessagesState)
     _builder.add_node("agent", call_model)
@@ -74,7 +78,9 @@ def draw_graph(graph: CompiledStateGraph) -> None:
     """
     Render the compiled state graph as a Mermaid PNG image and save it to the assets directory.
     """
-    graph.get_graph().draw_mermaid_png(output_file_path=ASSETS_DIR / "graph.png")
+    graph.get_graph().draw_mermaid_png(
+        output_file_path=ASSETS_DIR / "graph.png"
+    )
 
 
 if __name__ == "__main__":

src/chattr/gui.py

@@ -2,7 +2,9 @@ import gradio
 from gradio import Blocks, Button, Chatbot, ChatMessage, Row, Textbox
 
 
-def generate_response(history: list[ChatMessage], thread_id: str) -> list[ChatMessage]:
+def generate_response(
+    history: list[ChatMessage], thread_id: str
+) -> list[ChatMessage]:
     """
     Appends an assistant message about a quarterly sales plot to the chat history for the specified thread ID.
 
@@ -33,7 +35,9 @@ def app_block() -> Blocks:
 
     history = [
         ChatMessage(role="assistant", content="How can I help you?"),
-        ChatMessage(role="user", content="Can you make me a plot of quarterly sales?"),
+        ChatMessage(
+            role="user", content="Can you make me a plot of quarterly sales?"
+        ),
         ChatMessage(
             role="assistant",
             content="I am happy to provide you that report and plot.",
@@ -41,13 +45,17 @@ def app_block() -> Blocks:
     ]
     with Blocks() as app:
         with Row():
-            thread_id: Textbox = Textbox(label="Thread ID", info="Enter Thread ID")
+            thread_id: Textbox = Textbox(
+                label="Thread ID", info="Enter Thread ID"
+            )
 
         chatbot: Chatbot = Chatbot(history, type="messages")
 
         with Row():
             generate_btn: Button = Button(value="Generate", variant="primary")
             stop_btn: Button = Button(value="Stop", variant="stop")
-        _event = generate_btn.click(generate_response, [chatbot, thread_id], chatbot)
+        _event = generate_btn.click(
+            generate_response, [chatbot, thread_id], chatbot
+        )
         stop_btn.click(cancels=[_event])
     return app