Update app.py

app.py

@@ -60,7 +60,7 @@ except Exception as e:
     log_debug(f"❌ Model initialization failed: {str(e)}")
     raise
 
-def extract_text_from_pdf_url(url, debug_state):
+def extract_text_from_pdf_url(url):
     """Extract text from PDF with debug logging"""
     debug_state = log_debug(f"📄 Fetching PDF: {url[:60]}...")
     try:
@@ -77,24 +77,21 @@ def extract_text_from_pdf_url(url, debug_state):
         debug_state = log_debug(error_msg)
         return f"[Error loading PDF: {str(e)}]", debug_state
 
-def generate_answer(query, pdf_urls_str, debug_console):
+def generate_answer(query, pdf_urls_str):
     """Main processing function with debug output"""
     try:
         debug_state = log_debug(f"🔍 New query: {query}")
-        debug_console(debug_state)
 
         if not query or not pdf_urls_str:
             debug_state = log_debug("❌ Missing question or PDF URLs")
-            debug_console(debug_state)
-            return "Please provide both inputs"
+            return "Please provide both inputs", debug_state
 
         pdf_urls = [url.strip() for url in pdf_urls_str.strip().split("\n") if url.strip()]
         sources = []
         feedback = "### PDF Load Report:\n"
 
         for url in pdf_urls:
-            text, debug_state = extract_text_from_pdf_url(url, debug_state)
-            debug_console(debug_state)
+            text, debug_state = extract_text_from_pdf_url(url)
             if not text.startswith("[Error"):
                 # Create source document with metadata
                 sources.append({
@@ -107,11 +104,9 @@ def generate_answer(query, pdf_urls_str, debug_console):
 
         if not sources:
             debug_state = log_debug("❌ No valid PDFs processed")
-            debug_console(debug_state)
-            return feedback + "\nNo valid PDFs processed"
+            return feedback + "\nNo valid PDFs processed", debug_state
 
         debug_state = log_debug(f"🧠 Generating answer using {len(sources)} sources...")
-        debug_console(debug_state)
 
         try:
             # Generate without attention_mask parameter
@@ -121,25 +116,20 @@ def generate_answer(query, pdf_urls_str, debug_console):
             backend = response.get('backend_used', 'unknown')
 
             debug_state = log_debug(f"💡 Answer generated using {backend}")
-            debug_console(debug_state)
             full_output = f"{feedback}\n\n### Answer:\n{answer}\n\n_Generated using {backend}_"
-            return full_output
+            return full_output, debug_state
 
         except Exception as e:
             error_msg = f"❌ Generation error: {str(e)}"
             debug_state = log_debug(error_msg)
-            debug_console(debug_state)
             debug_state = log_debug(traceback.format_exc())
-            debug_console(debug_state)
-            return feedback + f"\n\n❌ Error: {str(e)}"
+            return feedback + f"\n\n❌ Error: {str(e)}", debug_state
 
     except Exception as e:
         error_msg = f"❌ System error: {str(e)}"
         debug_state = log_debug(error_msg)
-        debug_console(debug_state)
         debug_state = log_debug(traceback.format_exc())
-        debug_console(debug_state)
-        return error_msg
+        return error_msg, debug_state
 
 # Create the Gradio interface
 with gr.Blocks(title="Pleias RAG QA", css="""
@@ -172,20 +162,17 @@ with gr.Blocks(title="Pleias RAG QA", css="""
                     elem_classes=["debug-console"]
                 )
 
+    def update_debug_console():
+        return "\n".join(debug_messages)
+
     submit_btn.click(
         fn=generate_answer,
         inputs=[question, pdf_urls],
-        outputs=[output],
-        js="""
-        function updateDebugConsole(debug_state) {
-            const debugConsole = document.querySelector('.debug-console textarea');
-            if (debugConsole) {
-                debugConsole.value = debug_state;
-            }
-        }
-        """
+        outputs=[output, debug_console],
     )
 
+    demo.load(update_debug_console, None, debug_console, every=1)
+
 if __name__ == "__main__":
     log_debug("🚀 Launching interface...")
     demo.launch(