feat: use env var

.env.example

@@ -0,0 +1,2 @@
+LOG=debug
+TOKEN=

.gitignore

@@ -1,4 +1,5 @@
 venv
 .idea
 .DS_Store
-flagged
+flagged
+.env

app.py

@@ -1,4 +1,5 @@
 import logging
+import os
 import random
 
 import gradio as gr
@@ -66,24 +67,32 @@ def inference(file, prompt, quality, seed, version):
 interface = gr.Interface(
     fn=inference,
     inputs=[
-        gr.File(label="Upload Product Image"),
+        gr.File(label="Upload Garment Image"),
         gr.Textbox(label="Enter Prompt", placeholder="Enter your description here"),
         gr.Dropdown(
-            choices=["low", "high"], label="Quality", value="low", info="Select image quality"
-        ),
-        gr.Number(
-            label="Seed", value=random.randint(1, 2**64 - 1)
+            choices=["low", "high"],
+            label="Quality",
+            value="low",
+            info="Select image quality",
         ),
+        gr.Number(label="Seed", value=random.randint(1, 2**64 - 1)),
         gr.Dropdown(
             choices=["1"], label="Version", value="1", info="Select model version"
         ),
     ],
     outputs=gr.Gallery(label="Generated Images"),
-    title="Techhau",
+    title="Hautech",
     description="Upload a garment image and provide a prompt to generate related content.",
     theme="huggingface",
 )
 
 if __name__ == "__main__":
-    # logging.basicConfig(level=logging.DEBUG)
+    log_level = os.getenv("LOG")
+    if log_level is not None and log_level.lower() == "debug":
+        logging.basicConfig(level=logging.DEBUG)
+
+    token = os.getenv("TOKEN")
+    if not token:
+        raise Exception("Token environment variable is required")
+
     interface.launch()

hautech.py

@@ -1,9 +1,9 @@
 from __future__ import annotations
 
 import logging
+import os
 import random
 import time
-from os import PathLike
 import requests
 from typing import Optional, Dict
 from pydantic import BaseModel, Field
@@ -36,7 +36,7 @@ class HautechRequest:
     # todo: use env var
     def __init__(
         self,
-        token="eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCIsImtpZCI6ImRhODkzODdjLTRlMDEtNDNiZS04ODYxLTA3NzczYmQ4MmQ4YyJ9.eyJpc3MiOiIzY2I4Y2ZkYy00OGFjLTQ5ZGUtODU4ZC1mZmU4MGVmOTQ1NmQiLCJwZXJtaXNzaW9ucyI6eyJhY2Nlc3MiOnsiYWRkIjp0cnVlLCJyZWFkIjp0cnVlLCJyZW1vdmUiOnRydWV9LCJhY2NvdW50cyI6eyJjcmVhdGUiOmZhbHNlLCJyZWFkIjpmYWxzZX0sImJhbGFuY2VzIjp7InJlYWQiOnRydWUsInVwZGF0ZSI6ZmFsc2V9LCJjb2xsZWN0aW9ucyI6eyJjaGlsZHJlbiI6eyJhZGQiOnRydWUsInJlYWQiOnRydWUsInJlbW92ZSI6dHJ1ZX0sImNyZWF0ZSI6dHJ1ZSwicGFyZW50cyI6eyJhZGQiOnRydWUsInJlYWQiOnRydWUsInJlbW92ZSI6dHJ1ZX0sInJlYWQiOnRydWUsInN0YWNrcyI6eyJhZGQiOnRydWUsInJlbW92ZSI6dHJ1ZX0sInVwZGF0ZSI6dHJ1ZX0sImdyb3VwcyI6eyJhY2NvdW50cyI6eyJhZGQiOmZhbHNlLCJyZW1vdmUiOmZhbHNlfSwiY3JlYXRlIjpmYWxzZSwiZGVsZXRlIjpmYWxzZX0sImltYWdlcyI6eyJjcmVhdGUiOnRydWUsInJlYWQiOnRydWV9LCJvcGVyYXRpb25zIjp7ImNyZWF0ZSI6dHJ1ZSwicmVhZCI6dHJ1ZSwidXBkYXRlIjp0cnVlfSwic3RhY2tzIjp7ImNyZWF0ZSI6dHJ1ZSwib3BlcmF0aW9ucyI6eyJhZGQiOnRydWUsInJlbW92ZSI6dHJ1ZX0sInJlYWQiOnRydWUsInVwZGF0ZSI6dHJ1ZX0sInN0b3JhZ2UiOnsiY3JlYXRlIjp0cnVlLCJkZWxldGUiOnRydWUsInJlYWQiOnRydWUsInVwZGF0ZSI6dHJ1ZX19LCJzdWIiOiIiLCJpYXQiOjE3MzQwMTg2NDUsImV4cCI6MjY4MDcyNzIwNX0.fl0K5_xWeTIXzjQgjDZC9bHGWga4oNkDbPbKrEadYzLqVIWTZLNGCbE4SVuf1_SPbxrDXBs5Rrxza6ddRSIWXszOPNzQsG_zGdpOvlHyHcQF3aGPBVUaTr1j5pU8KHZXXMmI2MLNy23MEK7a9in6xsDFnPie6oJiTTae19O_p_j9KBaEhMRDl4kXxIH0XyjcflJnWyW2HOg6HaOxYyYHtVkV-ts5XNB3tMM9ekOb3MA5m9dgO0vrn1T8gJYWD6F-pRPYRbbjvwc54-yFnkSi9yEFejVw-oancKcK75-BAhL49Q-6EY9qwiGIP9Iwgxxh5E44RstNM-1pcjkZkusfGg",
+        token=os.getenv("TOKEN"),
     ):
         self._token: Optional[str] = token
         self._session = requests.Session()
@@ -45,6 +45,9 @@ class HautechRequest:
         self._initialize_session_headers()
 
     def _initialize_session_headers(self):
+        if not self._token:
+            raise RuntimeError("TOKEN environment variable is missing")
+
         self._session.headers.update(
             {
                 "Authorization": f"Bearer {self._token}",
@@ -65,7 +68,7 @@ class HautechRequest:
 
         return response
 
-    def upload_image(self, file: PathLike[str] | PathLike[bytes]) -> str:
+    def upload_image(self, file: os.PathLike[str] | os.PathLike[bytes]) -> str:
         url = f"{self.FILES_URL}/upload/image"
         response = self._session.request("PUT", url, files={"file": open(file, "rb")})
         if response.status_code < 200 or response.status_code >= 300: