PuristanLabs1 commited on
Commit
c952e90
·
verified ·
1 Parent(s): aeaf035

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -167,9 +167,13 @@ def extract_entities_with_stanza(text, chunk_size=1000):
167
  for chunk in chunks:
168
  doc = nlp(chunk)
169
  for ent in doc.ents:
170
- entities.append({"text": ent.text, "type": ent.type})
171
 
172
- return entities
 
 
 
 
173
 
174
  ### 4️⃣ TTS Functionality (KokoroTTS)
175
  @spaces.GPU(duration=1000)
@@ -276,7 +280,7 @@ with gr.Blocks() as demo:
276
  fetch_and_display_content,
277
  inputs=[url_input],
278
 
279
- outputs=[extracted_text, metadata_output, detected_lang, process_summary_button, process_audio_button, extracted_text, metadata_output]
280
  )
281
 
282
  process_summary_button.click(hierarchical_summarization, inputs=[extracted_text], outputs=[summary_output])
@@ -297,6 +301,5 @@ with gr.Blocks() as demo:
297
  outputs=[ner_output]
298
  )
299
 
300
- #process_button.click(process_url, inputs=[url_input], outputs=[extracted_text, detected_language, full_audio_output])
301
-
302
  demo.launch()
 
167
  for chunk in chunks:
168
  doc = nlp(chunk)
169
  for ent in doc.ents:
170
+ entities.append(f"📌 **Entity**: \"{ent.text}\" | **Type**: {ent.type}")
171
 
172
+ #return entities
173
+ if not entities:
174
+ return "No entities found."
175
+
176
+ return "\n\n".join(entities) # ✅ Display as Markdown-formatted text
177
 
178
  ### 4️⃣ TTS Functionality (KokoroTTS)
179
  @spaces.GPU(duration=1000)
 
280
  fetch_and_display_content,
281
  inputs=[url_input],
282
 
283
+ outputs=[extracted_text, metadata_output, detected_lang, process_summary_button, process_audio_button, summary_output, ner_output]
284
  )
285
 
286
  process_summary_button.click(hierarchical_summarization, inputs=[extracted_text], outputs=[summary_output])
 
301
  outputs=[ner_output]
302
  )
303
 
304
+
 
305
  demo.launch()