Update app.py

app.py

@@ -69,11 +69,11 @@ def analyze_chunk(chunk):
         )
         return stream.choices[0].message.content
     except Exception as e:
-        return f"Error while processing chunk: {e}"
+        return f"An error occurred while processing a chunk: {e}"
 
 
-def generate_docx(content):
-    """Generates a DOCX file from content."""
+def save_as_docx(content):
+    """Generates and saves a DOCX file."""
     document = Document()
     document.add_heading("Literature Analysis", level=1)
     document.add_paragraph(content)
@@ -83,24 +83,26 @@ def generate_docx(content):
 
 
 def analyze_document(file):
-    """Processes the document and generates insights."""
+    """Processes and analyzes the uploaded document."""
     text = extract_text_from_file(file)
     if text.startswith("Unsupported file format"):
-        return None, f"**Error:** {text}"
+        return "**Error:** Unsupported file format. Please upload a valid PDF or DOCX file.", None
 
     chunks = chunk_text(text)
     all_insights = []
-    markdown_output = ""
+    progress_output = ""
 
     for i, chunk in enumerate(chunks, 1):
+        progress_output += f"**Processing chunk {i} of {len(chunks)}...**\n"
         result = analyze_chunk(chunk)
         if result.strip():  # Only append non-empty results
             all_insights.append(result)
-            markdown_output += f"### Chunk {i} Analysis\n{result}\n\n"
+        progress_output += f"**Chunk {i} Analysis Complete:**\n{result}\n\n"
 
     if not all_insights:
-        return None, "**Error:** No valid insights were extracted from the document."
+        return "**Error:** No valid insights were extracted from the document.", None
 
+    # Consolidate final summary
     consolidated_summary_prompt = (
         "Below are insights extracted from multiple chunks. "
         "Consolidate these insights into a single output organized as follows: "
@@ -122,37 +124,35 @@ def analyze_document(file):
         )
         final_summary = ""
         for chunk in stream:
-            final_summary += chunk.choices[0].delta.content or ""
+            content = chunk.choices[0].delta.content or ""
+            final_summary += content
 
-        # Return the final summary for display, and the content for DOCX generation
-        return markdown_output + f"\n\n### Final Summary\n\n{final_summary}", final_summary
+        progress_output += f"**Final Consolidated Summary:**\n\n{final_summary}"
+
+        # Generate DOCX file after processing
+        docx_file = save_as_docx(final_summary)
+        return progress_output, docx_file
     except Exception as e:
-        return None, f"**Error:** An error occurred during consolidation: {e}"
+        return f"**Error:** An error occurred during consolidation: {e}", None
 
 
+# Define Gradio interface
 def interface_logic(file):
-    """Handles the Gradio interface logic."""
-    markdown_output, docx_content = analyze_document(file)
-    if docx_content:
-        # Generate the DOCX file after analysis is complete
-        docx_file = generate_docx(docx_content)
-        return markdown_output, docx_file
-    else:
-        return markdown_output, None
+    markdown_output, docx_file = analyze_document(file)
+    return markdown_output, docx_file
 
 
-# Define Gradio interface
 interface = gr.Interface(
     fn=interface_logic,
     inputs=gr.File(label="Upload a PDF or DOCX file"),
     outputs=[
-        gr.Markdown(label="Literature Analysis"),
+        gr.Markdown(label="Progress and Analysis"),
         gr.File(label="Download Analysis as DOCX")
     ],
     title="Automated Literature Review",
     description=(
-        "Upload a PDF or DOCX document, and this tool will analyze it to extract and consolidate its content. "
-        "Progress updates will be shown during processing. After analysis, you can download the report as a DOCX file."
+        "Upload a PDF or DOCX document. The tool will analyze it chunk by chunk, display progress, and generate a final summary. "
+        "You can download the consolidated report as a DOCX file after processing."
     ),
 )