Update app.py

app.py

@@ -83,17 +83,20 @@ def analyze_document(file):
     """Processes and analyzes the uploaded document."""
     text = extract_text_from_file(file)
     if text.startswith("Unsupported file format"):
-        return text
+        yield f"**Error:** {text}"
+        return
 
     chunks = chunk_text(text)
     all_insights = []
 
-    for chunk in chunks:
+    yield "**Processing the document. Please wait...**\n"
+    for i, chunk in enumerate(chunks, 1):
+        yield f"**Processing chunk {i} of {len(chunks)}...**"
         result = analyze_chunk(chunk)
         if result.strip():  # Only append non-empty results
             all_insights.append(result)
 
-    # Consolidate all results into a single final summary
+    yield "**Consolidating all insights into a final summary...**"
     consolidated_summary_prompt = (
         "Below are insights extracted from multiple chunks of a document. "
         "Consolidate these insights into a single output organized as follows: "
@@ -112,7 +115,7 @@ def analyze_document(file):
         stream=False
     )
 
-    return final_summary.choices[0].message.content
+    yield f"**Final Summary:**\n\n{final_summary.choices[0].message.content}"
 
 # Define the Gradio interface
 interface = gr.Interface(
@@ -128,4 +131,4 @@ interface = gr.Interface(
 
 # Launch the interface
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()