add listen

listen.py

@@ -23,21 +23,6 @@ except Exception:
 # ---------- Blueprint ----------
 listen_bp = Blueprint("listen", __name__)
 
-# Directories for video, audio, and transcripts (local working paths)
-VIDEO_FOLDER = 'static/videos'
-AUDIO_FOLDER = 'static/audio'
-TRANSCRIPT_FOLDER = 'static/transcripts'
-
-# Ensure local working dirs exist
-os.makedirs(VIDEO_FOLDER, exist_ok=True)
-os.makedirs(AUDIO_FOLDER, exist_ok=True)
-os.makedirs(TRANSCRIPT_FOLDER, exist_ok=True)
-
-# ---------------- Cohere configuration (migrated to v2 Chat) ----------------
-COHERE_API_KEY = os.getenv("COHERE_API_KEY", "")
-COHERE_API_URL = 'https://api.cohere.com/v2/chat'
-# ---------------------------------------------------------------------------
-
 # ---------------------- storage mode helpers ----------------------
 def _is_aws_video_mode() -> bool:
     """
@@ -70,7 +55,36 @@ def _video_s3_key(filename: str) -> str:
     prefix = prefix.strip().strip("/")
     return f"{prefix}/{filename}"
 
-# -------------------------------------------------------------------
+# ---------- writable working directories ----------
+# Base working dir: /tmp on HF/AWS; local stays under ./static (or override via LISTEN_WORKDIR)
+_BASE_WORKDIR = os.getenv(
+    "LISTEN_WORKDIR",
+    "/tmp/listen" if _is_aws_video_mode() else os.path.abspath("static")
+)
+
+VIDEO_FOLDER = os.path.join(_BASE_WORKDIR, "videos")
+AUDIO_FOLDER = os.path.join(_BASE_WORKDIR, "audio")
+TRANSCRIPT_FOLDER = os.path.join(_BASE_WORKDIR, "transcripts")
+
+# Ensure directories exist (with hard fallback to /tmp if needed)
+for _pname in ("videos", "audio", "transcripts"):
+    _p = os.path.join(_BASE_WORKDIR, _pname)
+    try:
+        os.makedirs(_p, exist_ok=True)
+    except Exception:
+        _fallback_base = "/tmp/listen"
+        os.makedirs(os.path.join(_fallback_base, _pname), exist_ok=True)
+        if _pname == "videos":
+            VIDEO_FOLDER = os.path.join(_fallback_base, "videos")
+        elif _pname == "audio":
+            AUDIO_FOLDER = os.path.join(_fallback_base, "audio")
+        else:
+            TRANSCRIPT_FOLDER = os.path.join(_fallback_base, "transcripts")
+
+# ---------------- Cohere configuration (migrated to v2 Chat) ----------------
+COHERE_API_KEY = os.getenv("COHERE_API_KEY", "")
+COHERE_API_URL = 'https://api.cohere.com/v2/chat'
+# ---------------------------------------------------------------------------
 
 # Google Cloud Speech-to-Text Configuration
 speech_client = speech.SpeechClient()
@@ -181,7 +195,7 @@ def serve_video(filename):
 @listen_bp.route('/upload-video', methods=['POST'])
 def upload_video():
     """
-    Local: save to static/videos.
+    Local: save to static/videos or /tmp/listen/videos (depending on mode).
     HF/AWS: upload to S3 (no local original).
     """
     print("Received upload request.")