Update app.py

app.py

@@ -65,17 +65,15 @@ def fetch_and_display_content(url):
     #return cleaned_text, metadata, detected_lang, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 	
     return (
-        cleaned_text,         # ✅ Extracted content
-        metadata,             # ✅ Article metadata
-        detected_lang,        # ✅ Detected language
-        gr.update(visible=True),  # ✅ Show Summary button
-        gr.update(visible=True),  # ✅ Show Audio button
+        cleaned_text,          # ✅ Extracted text
+        metadata,              # ✅ Metadata (must be a valid JSON object)
+        detected_lang,         # ✅ Detected language
+        gr.update(visible=True),  # ✅ Make Summary button visible
+        gr.update(visible=True),  # ✅ Make Audio button visible
         gr.update(visible=True),  # ✅ Show Extracted text box
         gr.update(visible=True),  # ✅ Show Metadata box
-        "",                    # ✅ Reset Summary output when a new URL is fetched
-        "",                    # ✅ Reset Entity output when a new URL is fetched
-        gr.update(value=cleaned_text, visible=True),  # ✅ Ensure Extracted Text is shown
-        gr.update(value=metadata, visible=True)       # ✅ Ensure Metadata is shown
+        json.dumps([]),         # ✅ Reset Entity output with an empty JSON array instead of ""
+        ""                      # ✅ Reset Summary output
     )
 
 ### 2️⃣ Cleaning Function
@@ -180,14 +178,10 @@ def extract_entities_with_stanza(text, chunk_size=1000):
     for chunk in chunks:
         doc = nlp(chunk)
         for ent in doc.ents:
-            
-            entities.append(f"📌 **Entity**: \"{ent.text}\"  |  **Type**: {ent.type}")  # ✅ Format output
+            entities.append({"text": ent.text, "type": ent.type})
 
-    #return entities
-    if not entities:
-        return "No entities found."
-
-    return "\n\n".join(entities)  # ✅ Display as Markdown-formatted text
+    # ✅ Ensure JSON format even if no entities are found
+    return json.dumps({"entities": entities}) if entities else json.dumps({"entities": []})
 
 ### 4️⃣ TTS Functionality (KokoroTTS)
 @spaces.GPU(duration=1000)  
@@ -295,10 +289,11 @@ with gr.Blocks() as demo:
         inputs=[url_input], 
         
         outputs=[
-            extracted_text, metadata_output, detected_lang, 
-            process_summary_button, process_audio_button, 
-            summary_output, ner_output,
-            extracted_text, metadata_output  # ✅ Ensures visibility update
+        extracted_text, metadata_output, detected_lang, 
+        process_summary_button, process_audio_button, 
+        summary_output, ner_output,  # ✅ Existing Outputs
+        gr.update(visible=True),  # ✅ Ensures Extracted Text Box is shown
+        gr.update(visible=True)   # ✅ Ensures Metadata Box is shown
        ]
     )