fix

main.py

@@ -517,92 +517,85 @@ def main():
         margin-bottom: 16px !important;
     }
     """
-    js = r'''
-const uploadMessages = [
-  "Crunching your documents...",
-  "Warming up the AI...",
-  "Extracting knowledge...",
-  "Scanning for insights...",
-  "Preparing your data...",
-  "Looking for answers...",
-  "Analyzing file structure...",
-  "Reading your files...",
-  "Indexing content...",
-  "Almost ready..."
-];
-
-let msgInterval = null;
-let timerInterval = null;
-let startMs = 0;
-let lastMsg = null;
-
-function root() {
-  return document.getElementById("processing-message");
-}
-function isVisible(el) {
-  return !!(el && (el.offsetWidth || el.offsetHeight || el.getClientRects().length));
-}
-function pickMsg() {
-  if (uploadMessages.length === 0) return "";
-  if (uploadMessages.length === 1) return uploadMessages[0];
-  let m;
-  do { m = uploadMessages[Math.floor(Math.random() * uploadMessages.length)]; }
-  while (m === lastMsg);
-  lastMsg = m;
-  return m;
-}
-function getMsgSpan() {
-  const r = root();
-  return r ? r.querySelector("#processing-msg") : null;
-}
-function getTimerSpan() {
-  const r = root();
-  return r ? r.querySelector("#processing-timer") : null;
-}
-function setMsg(t) {
-  const s = getMsgSpan();
-  if (s) s.textContent = t;
-}
-function fmtElapsed() {
-  return ((Date.now() - startMs) / 1000).toFixed(1) + "s elapsed";
-}
+    js = r"""
+(function () {
+  var uploadMessages = [
+    "Crunching your documents...",
+    "Warming up the AI...",
+    "Extracting knowledge...",
+    "Scanning for insights...",
+    "Preparing your data...",
+    "Looking for answers...",
+    "Analyzing file structure...",
+    "Reading your files...",
+    "Indexing content...",
+    "Almost ready..."
+  ];
+
+  var msgInterval = null;
+  var timerInterval = null;
+  var startMs = 0;
+  var lastMsg = null;
+
+  function root() { return document.getElementById("processing-message"); }
+  function isVisible(el) { return !!(el && (el.offsetWidth || el.offsetHeight || el.getClientRects().length)); }
+
+  function pickMsg() {
+    if (uploadMessages.length === 0) return "";
+    if (uploadMessages.length === 1) return uploadMessages[0];
+    var m;
+    do { m = uploadMessages[Math.floor(Math.random() * uploadMessages.length)]; }
+    while (m === lastMsg);
+    lastMsg = m;
+    return m;
+  }
+
+  function getMsgSpan() {
+    return document.getElementById("processing-msg") || (root() ? root().querySelector("#processing-msg") : null);
+  }
+  function getTimerSpan() {
+    return document.getElementById("processing-timer") || (root() ? root().querySelector("#processing-timer") : null);
+  }
+
+  function setMsg(t) { var s = getMsgSpan(); if (s) s.textContent = t; }
+  function fmtElapsed() { return ((Date.now() - startMs) / 1000).toFixed(1) + "s elapsed"; }
 
-function start() {
-  if (msgInterval || timerInterval) return;
-  startMs = Date.now();
-  setMsg(pickMsg());
+  function start() {
+    if (msgInterval || timerInterval) return;
+    startMs = Date.now();
+    setMsg(pickMsg());
 
-  msgInterval = setInterval(() => setMsg(pickMsg()), 2000);
+    msgInterval = setInterval(function(){ setMsg(pickMsg()); }, 2000);
 
-  const t = getTimerSpan();
-  if (t) {
-    t.textContent = fmtElapsed();
-    timerInterval = setInterval(() => { t.textContent = fmtElapsed(); }, 200);
+    var t = getTimerSpan();
+    if (t) {
+      t.textContent = fmtElapsed();
+      timerInterval = setInterval(function(){ t.textContent = fmtElapsed(); }, 200);
+    }
   }
-}
 
-function stop() {
-  if (msgInterval) { clearInterval(msgInterval); msgInterval = null; }
-  if (timerInterval) { clearInterval(timerInterval); timerInterval = null; }
-  const t = getTimerSpan();
-  if (t) t.textContent = "";
-}
+  function stop() {
+    if (msgInterval) { clearInterval(msgInterval); msgInterval = null; }
+    if (timerInterval) { clearInterval(timerInterval); timerInterval = null; }
+    var t = getTimerSpan();
+    if (t) t.textContent = "";
+  }
 
-function tick() {
-  const r = root();
-  if (isVisible(r)) start();
-  else stop();
-}
+  function tick() {
+    var r = root();
+    if (isVisible(r)) start();
+    else stop();
+  }
 
-// Observe rerenders / visibility changes
-const obs = new MutationObserver(tick);
-obs.observe(document.body, { subtree: true, childList: true, attributes: true });
+  var obs = new MutationObserver(tick);
+  obs.observe(document.body, { subtree: true, childList: true, attributes: true });
 
-window.addEventListener("load", tick);
-setInterval(tick, 500);
-'''
+  window.addEventListener("load", tick);
+  setInterval(tick, 500);
+})();
+"""
 
-    with gr.Blocks(theme=gr.themes.Soft(), title="SmartDoc AI", css=css, js=js) as demo:
+    with gr.Blocks(theme=gr.themes.Soft(), title="SmartDoc AI", css=css) as demo:
         gr.Markdown("### SmartDoc AI - Document Q&A", elem_classes="app-title")
         gr.Markdown("Upload your documents and ask questions. Answers will appear below, just like a chat.", elem_classes="app-description")
         gr.Markdown("---")
@@ -888,19 +881,19 @@ setInterval(tick, 500);
     # Launch server - Compatible with both local and Hugging Face Spaces
     # HF Spaces sets SPACE_ID environment variable
     is_hf_space = os.environ.get("SPACE_ID") is not None
-    
     demo.queue()
     if is_hf_space:
         # Hugging Face Spaces configuration
         logger.info("Running on Hugging Face Spaces")
-        demo.launch(server_name="0.0.0.0", server_port=7860)
+        demo.launch(server_name="0.0.0.0", server_port=7860, js=js)
     else:
         # Local development configuration
         configured_port = int(os.environ.get("GRADIO_SERVER_PORT", "7860"))
         server_port = _find_open_port(configured_port)
         logger.info(f"Launching Gradio on port {server_port}")
         logger.info(f"Access the app at: http://127.0.0.1:{server_port}")
-        demo.launch(server_name="127.0.0.1", server_port=server_port, share=False)
+        demo.launch(server_name="127.0.0.1", server_port=server_port, share=False, js=js)
+
 
 
 if __name__ == "__main__":