Update streamlit_app.py

streamlit_app.py

@@ -21,7 +21,8 @@ import os
 import string
 import traceback
 from pathlib import Path
-from typing import List, Tuple, Optional
+from typing import Tuple, Optional
+from difflib import SequenceMatcher  # <-- needed for prompt‑echo stripping
 
 # ----------------------------------------------------------------------
 # Third‑party libraries
@@ -71,7 +72,7 @@ DEFAULT_STATE = {
     "video_path": "",
     "model_input": DEFAULT_MODEL,
     "prompt": DEFAULT_PROMPT,
-    "api_key": os.getenv("GOOGLE_API_KEY", ""),
+    "api_key": os.getenv("GOOGLE_API_KEY", "AIzaSyBiAW2GQLid0HGe9Vs_ReKwkwsSVNegNzs"),
     "video_password": "",
     "compress_mb": 200,
     "busy": False,
@@ -117,7 +118,6 @@ def _convert_to_mp4(src: Path) -> Path:
     except ffmpeg.Error as e:
         raise RuntimeError(f"ffmpeg conversion failed: {e.stderr.decode()}") from e
 
-    # Remove original if conversion succeeded
     if dst.exists() and dst.stat().st_size > 0:
         src.unlink()
     return dst
@@ -168,13 +168,11 @@ def _download_with_yt_dlp(url: str, dst: Path, password: str = "") -> Path:
     except Exception as e:
         raise RuntimeError(f"yt‑dlp could not download the URL: {e}") from e
 
-    # Predictable filename from yt‑dlp info dict
     if isinstance(info, dict) and "id" in info:
         candidate = dst / f"{info['id']}.{info.get('ext', 'mp4')}"
         if candidate.exists():
             return _convert_to_mp4(candidate)
 
-    # Fallback: newest file in the folder
     files = list(dst.iterdir())
     if not files:
         raise RuntimeError("yt‑dlp did not produce any files.")
@@ -193,11 +191,9 @@ def download_video(url: str, dst: Path, password: str = "") -> Path:
     """
     video_exts = (".mp4", ".mov", ".webm", ".mkv", ".avi", ".flv")
 
-    # 1️⃣ Direct file
     if url.lower().endswith(video_exts):
         return _download_direct(url, dst)
 
-    # 2️⃣ Twitter
     if "twitter.com" in url and "/status/" in url:
         tweet_id = url.split("/")[-1].split("?")[0]
         for tweet in sntwitter.TwitterTweetScraper(tweet_id).get_items():
@@ -209,7 +205,7 @@ def download_video(url: str, dst: Path, password: str = "") -> Path:
                     return download_video(u.expandedUrl, dst)
         raise RuntimeError("No video found in the tweet.")
 
-    # 3️⃣ yt‑dlp
+    # Fallback to yt‑dlp for any other URL
     return _download_with_yt_dlp(url, dst, password)
 
 
@@ -237,7 +233,7 @@ def generate_report(
     return getattr(resp, "text", str(resp))
 
 
-def _strip_prompt_echo(prompt: str, text: str, threshold: float = 0
+def _strip_prompt_echo(prompt: str, text: str, threshold: float = 0.68) -> str:
     """Remove the prompt if the model repeats it at the start of *text*."""
     if not prompt or not text:
         return text