Upload app.py

app.py

@@ -1,5 +1,7 @@
 # app.py — Streamlit Frontend for Diag Studio AI
-import os, json, time, io, requests, traceback
+import os
+import json
+import requests
 import streamlit as st
 
 # ---- container-safe defaults to avoid '/.streamlit' permission errors
@@ -15,8 +17,6 @@ try:
     st._config.set_option("theme.secondaryBackgroundColor", "#171a1f")
     st._config.set_option("theme.textColor", "#e6e8ea")
 except Exception:
-    # Themes usually set via config.toml, but programmatic options apply at runtime.
-    # See Streamlit theming docs.  # docs: https://docs.streamlit.io/develop/concepts/configuration/theming
     pass
 
 # ================= Env & API =================
@@ -115,7 +115,7 @@ backend_default = API_BASE or "http://127.0.0.1:7860"
 api_base = st.text_input(t["backend_url"], value=backend_default)
 
 # Health check
-health_col, spacer = st.columns([0.28, 0.72])
+health_col, _ = st.columns([0.28, 0.72])
 with health_col:
     if st.button("🔄 Health"):
         try:
@@ -131,8 +131,45 @@ st.markdown("---")
 
 # Upload or URL input
 st.write(t["upload_hint"])
-file = st.file_uploader(t["upload"], type=["mp3", "wav", "ogg"])  # accepts file-like objects; returns UploadedFile. Docs show in-memory behavior.  # 
-        files = {"file": (_file.name, _file, _file.type or "application/octet-stream")}
+file = st.file_uploader(t["upload"], type=["mp3", "wav", "ogg"])
+st.write(f"**{t['or']}**")
+audio_url = st.text_input(t["paste_url"], value="", placeholder="https://...")
+
+with st.expander(t["adv"], expanded=False):
+    detailed = st.checkbox(t["detailed"], value=True)
+    use_data_uri = st.checkbox(t["use_data_uri"], value=True)
+    openai_model = st.text_input(t["model"], value="gpt-4o")
+    prompt_override = st.text_area(
+        t["prompt"],
+        value="",
+        height=140,
+        placeholder="Leave empty to use backend default prompt",
+    )
+    st.caption(t["footer"])
+
+c1, c2 = st.columns([0.25, 0.75])
+go = c1.button(f"▶ {t['analyze']}", type="primary")
+reset = c2.button(f"↩ {t['reset']}")
+
+if reset:
+    st.session_state.last_response = None
+    st.rerun()
+
+# ============== Submission ==============
+def post_analyze(_api_base: str, _headers: dict, _file, _audio_url: str) -> dict:
+    url = f"{_api_base}/analyze"
+    data = {
+        "detailed_analysis": "true" if detailed else "false",
+        "use_data_uri": "true" if use_data_uri else "false",
+        "openai_model": openai_model or "gpt-4o",
+    }
+    if prompt_override.strip():
+        data["prompt"] = prompt_override.strip()
+
+    files = None
+    if _file is not None:
+        mime = getattr(_file, "type", None) or "application/octet-stream"
+        files = {"file": (_file.name, _file, mime)}
     elif _audio_url.strip():
         data["audio_url"] = _audio_url.strip()
 
@@ -169,7 +206,7 @@ if res:
     parsed = res.get("fal_parsed")
     if parsed:
         st.markdown(f"**{t['json']}**")
-        st.json(parsed)  # renders a collapsible JSON tree
+        st.json(parsed)
 
     # Backend artifacts — download via Streamlit
     st.markdown(f"**{t['backend_artifacts']}**")
@@ -177,7 +214,10 @@ if res:
     col_a, col_b, col_c = st.columns(3)
 
     def fetch_bytes(url: str) -> bytes:
-        r = requests.get(url if url.startswith("http") else f"{api_base}{url}", headers=HEADERS, timeout=60)
+        if not url:
+            raise RuntimeError("empty url")
+        full = url if url.startswith("http") else f"{api_base}{url}"
+        r = requests.get(full, headers=HEADERS, timeout=60)
         r.raise_for_status()
         return r.content
 
@@ -186,15 +226,17 @@ if res:
         try:
             data = fetch_bytes(dls["summary_txt"])
             col_a.download_button(label=t["download_summary"], data=data, file_name="summary.txt")
-        except Exception as e:
+        except Exception:
             col_a.error("summary.txt ✖")
 
     # parsed
     if dls.get("fal_parsed_json"):
         try:
             data = fetch_bytes(dls["fal_parsed_json"])
-            col_b.download_button(label=t["download_parsed"], data=data, file_name="parsed.json", mime="application/json")
-        except Exception as e:
+            col_b.download_button(
+                label=t["download_parsed"], data=data, file_name="parsed.json", mime="application/json"
+            )
+        except Exception:
             col_b.error("parsed.json ✖")
 
     # raw
@@ -202,7 +244,7 @@ if res:
         try:
             data = fetch_bytes(dls["fal_raw_txt"])
             col_c.download_button(label=t["download_raw"], data=data, file_name="raw.txt")
-        except Exception as e:
+        except Exception:
             col_c.error("raw.txt ✖")
 
 # ===== Footer =====