Update app.py

app.py

@@ -188,8 +188,10 @@ def generate_final_summary(agent, combined: str) -> str:
     combined = remove_duplicate_paragraphs(combined)
     final_prompt = f"""
 You are an expert clinical summarizer. Analyze the following summaries carefully and generate a **single final concise structured medical report**, avoiding any repetition or redundancy.
+
 Summaries:
 {combined}
+
 Respond with:
 - Diagnostic Patterns
 - Medication Issues
@@ -227,6 +229,7 @@ def process_report(agent, file, messages: List[Dict[str, str]]) -> Tuple[List[Di
         messages.append({"role": "assistant", "content": "❌ Please upload a valid file."})
         return messages, None
 
+    start_time = time.time()  # Start timing here
     messages.append({"role": "user", "content": f"📂 Processing file: {os.path.basename(file.name)}"})
     try:
         extracted = extract_text(file.name)
@@ -246,18 +249,24 @@ def process_report(agent, file, messages: List[Dict[str, str]]) -> Tuple[List[Di
             return messages, None
 
         summary = generate_final_summary(agent, "\n\n".join(valid))
+
         report_path = os.path.join(report_dir, f"report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md")
         with open(report_path, 'w', encoding='utf-8') as f:
             f.write(f"# 🧠 Final Medical Report\n\n{summary}")
 
-        messages.append({"role": "assistant", "content": f"📊 Final Report:\n\n{summary}"})
-        messages.append({"role": "assistant", "content": f"✅ Report saved: {os.path.basename(report_path)}"})
+        end_time = time.time()
+        elapsed_time = end_time - start_time
+
+        print(f"✅ Total processing time: {elapsed_time:.2f} seconds")  # Print the time
+
+        messages.append({"role": "assistant", "content": f"📊 Final Report generated in **{elapsed_time:.2f} seconds**.\n\n✅ Report saved: {os.path.basename(report_path)}"})
         return messages, report_path
 
     except Exception as e:
         messages.append({"role": "assistant", "content": f"❌ Error: {str(e)}"})
         return messages, None
 
+
 def create_ui(agent):
     with gr.Blocks(css="""
         html, body, .gradio-container { background: #0e1621; color: #e0e0e0; padding: 16px; }