Update app.py

Made the Huggingface version filename-agnostic.

app.py

@@ -1,74 +1,133 @@
 import sys
+import shutil
 from pathlib import Path
+from datetime import datetime
+
+import gradio as gr
 
+# --- Ensure src/ is importable (HF-only fix for src layout) ---
 ROOT = Path(__file__).resolve().parent
 SRC = ROOT / "src"
-
 sys.path.insert(0, str(SRC))
 
-import gradio as gr
-from pathlib import Path
-import shutil
-
 from ig_story_audit.main import main as run_audit
+from ig_story_audit.utils.config import load_config
 
 
-UPLOAD_DIR = Path("data/raw_stories")
-UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
+RAW_DIR = Path("data/raw_stories")
+RAW_DIR.mkdir(parents=True, exist_ok=True)
 
 
-def audit_stories(files, goal, language):
-    # Clear previous uploads
-    for f in UPLOAD_DIR.iterdir():
+def normalise_uploaded_files(files, run_date: str):
+    """
+    Hugging Face–only preprocessing:
+    - Clears any existing files for the run date
+    - Renames uploaded files into YYYY-MM-DD_story_N.ext
+    """
+    # Remove existing files for this date
+    for f in RAW_DIR.glob(f"{run_date}_story_*"):
         f.unlink()
 
-    # Save uploaded files
-    for file in files:
-        shutil.copy(file.name, UPLOAD_DIR / Path(file.name).name)
+    for idx, file in enumerate(files, start=1):
+        src = Path(file.name)
+        ext = src.suffix.lower() or ".jpg"
+        target = RAW_DIR / f"{run_date}_story_{idx}{ext}"
+        shutil.copy(src, target)
+
+
+def run_hf_audit(files, goal_text, language, is_today, selected_date):
+    if not files:
+        return "No files uploaded."
+
+    # Determine run date
+    if is_today:
+        run_date = datetime.now().strftime("%Y-%m-%d")
+    else:
+        run_date = selected_date.strip()
 
-    # Override config dynamically (simple approach)
-    from ig_story_audit.utils.config import load_config
+    # Prepare files
+    normalise_uploaded_files(files, run_date)
+
+    # Override language in config (runtime-only)
     config = load_config()
+    if "general" not in config:
+        config["general"] = {}
     config["general"]["language"] = language
 
-    # Run audit
+    # Run core pipeline (unchanged)
     run_audit()
 
-    # Read latest audit output
-    audit_dir = Path("data/audit_logs")
-    latest = max(audit_dir.glob("*_narrative_audit.txt"), key=lambda p: p.stat().st_mtime)
+    # Fetch latest audit output
+    audit_dir = Path(config["paths"]["audit_logs_dir"])
+    if not audit_dir.exists():
+        return "Audit completed, but no audit logs were generated."
+
+    latest = max(
+        audit_dir.glob("*_narrative_audit.txt"),
+        key=lambda p: p.stat().st_mtime,
+        default=None,
+    )
+
+    if not latest:
+        return "Audit completed, but no narrative audit file found."
 
     return latest.read_text(encoding="utf-8")
 
 
+# ----------------- Gradio UI -----------------
+
 with gr.Blocks() as demo:
-    gr.Markdown("## Instagram Story Narrative Audit")
+    gr.Markdown("## Instagram Story Narrative Audit (Manual Upload)")
+    gr.Markdown(
+        "Upload Instagram Story images manually. "
+        "Files will be analysed as a story sequence for the selected date."
+    )
 
-    files = gr.File(
+    files_input = gr.File(
         file_types=["image"],
         file_count="multiple",
-        label="Upload Instagram Story images (JPG/PNG)",
+        label="Upload Story Images",
     )
 
-    goal = gr.Textbox(
-        label="Audit goal",
+    goal_input = gr.Textbox(
+        label="Narrative goal",
         placeholder="e.g. Reinforce the feeling of tiredness through repetition",
     )
 
-    language = gr.Dropdown(
+    language_input = gr.Dropdown(
         choices=["id", "en"],
         value="id",
         label="Output language",
     )
 
-    run_btn = gr.Button("Run Audit")
+    is_today_input = gr.Checkbox(
+        value=True,
+        label="Use today's date",
+    )
+
+    date_input = gr.Textbox(
+        label="Date (YYYY-MM-DD)",
+        placeholder="2025-12-14",
+        visible=True,
+    )
 
-    output = gr.Textbox(label="Audit Result", lines=20)
+    run_button = gr.Button("Run Audit")
+
+    output_box = gr.Textbox(
+        label="Audit Result",
+        lines=25,
+    )
 
-    run_btn.click(
-        audit_stories,
-        inputs=[files, goal, language],
-        outputs=output,
+    run_button.click(
+        fn=run_hf_audit,
+        inputs=[
+            files_input,
+            goal_input,
+            language_input,
+            is_today_input,
+            date_input,
+        ],
+        outputs=output_box,
     )
 
 demo.launch()