Update app.py

app.py

@@ -26,16 +26,7 @@ def load_model():
         print(f"❌ Model loading failed: {str(e)}")
         return None
 
-# 2. Warm Start Mechanism
-def keep_model_warm():
-    """Periodic ping to prevent Hugging Face from unloading the model"""
-    if extractor:
-        try:
-            extractor("ping", max_length=1)
-        except:
-            pass
-
-# 3. Processing Function with Streamed Output
+# 2. Processing Function with Streamed Output
 def extract_structure(template, text):
     # Input validation
     if not text.strip():
@@ -71,7 +62,7 @@ def extract_structure(template, text):
         )[0]['generated_text']
         
         # Format output
-        formatted_json = json.dumps(json.loads(result), indent=2)
+        formatted_json = json.loads(result)  # Parse to validate JSON
         html_output = f"""
         <div style='
             padding: 15px;
@@ -81,7 +72,7 @@ def extract_structure(template, text):
             margin-top: 10px;
         '>
             <h3 style='margin-top:0'>Extracted Data</h3>
-            <pre style='white-space: pre-wrap'>{formatted_json}</pre>
+            <pre style='white-space: pre-wrap'>{json.dumps(formatted_json, indent=2)}</pre>
         </div>
         """
         
@@ -91,7 +82,7 @@ def extract_structure(template, text):
         error_msg = f"❌ Processing error: {str(e)}"
         yield error_msg, "", f"<p style='color:red'>{error_msg}</p>"
 
-# 4. Gradio Interface
+# 3. Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft(), title="NuExtract 1.5") as demo:
     # Header
     gr.Markdown("""
@@ -136,13 +127,9 @@ with gr.Blocks(theme=gr.themes.Soft(), title="NuExtract 1.5") as demo:
             gr.Markdown("### 📤 Results")
             status = gr.Textbox(
                 label="Status",
-                value="🟢 System Ready",
-                interactive=False
-            )
-            json_output = gr.JSON(
-                label="Structured Output",
-                interactive=False
+                value="🟢 System Ready"
             )
+            json_output = gr.JSON(label="Structured Output")  # Removed interactive parameter
             html_output = gr.HTML(
                 label="Formatted View",
                 value="<div style='min-height:200px'></div>"
@@ -160,28 +147,19 @@ with gr.Blocks(theme=gr.themes.Soft(), title="NuExtract 1.5") as demo:
     )
     
     clear_btn.click(
-        fn=lambda: ["", "", "", "<div></div>"],
+        fn=lambda: ["", "", {}, "<div></div>"],
         inputs=[],
         outputs=[template_input, text_input, json_output, html_output]
     )
 
-# 5. Launch Configuration
+# 4. Launch Configuration
 if __name__ == "__main__":
     # Initialize model
     extractor = load_model()
     
-    # Start keep-alive thread
-    import threading
-    threading.Thread(
-        target=lambda: [keep_model_warm() for _ in iter(int, 1)],
-        daemon=True
-    ).start()
-    
     # Launch app
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        show_error=True,
-        share=False,
-        favicon_path="https://huggingface.co/favicon.ico"
+        show_error=True
     )