Update app.py

app.py

@@ -197,7 +197,8 @@ async def process_final_report(agent, file, chatbot_state: List[Dict[str, str]])
 
     if file is None or not hasattr(file, "name"):
         messages.append({"role": "assistant", "content": "❌ Please upload a valid Excel file before analyzing."})
-        return messages, report_path
+        yield messages, report_path
+        return
 
     try:
         messages.append({"role": "user", "content": f"Processing Excel file: {os.path.basename(file.name)}"})
@@ -233,7 +234,8 @@ async def process_final_report(agent, file, chatbot_state: List[Dict[str, str]])
         chunk_responses = [r for r in chunk_responses if r]
         if not chunk_responses:
             messages.append({"role": "assistant", "content": "❌ No valid chunk responses to summarize."})
-            return messages, report_path
+            yield messages, report_path
+            return
 
         # Summarize chunk responses incrementally
         summary = ""
@@ -265,7 +267,8 @@ async def process_final_report(agent, file, chatbot_state: List[Dict[str, str]])
                     current_summary_tokens = estimate_tokens(summary)
                 except Exception as e:
                     messages.append({"role": "assistant", "content": f"❌ Error summarizing intermediate results: {str(e)}"})
-                    return messages, report_path
+                    yield messages, report_path
+                    return
 
             summary += f"\n\n### Chunk {i+1} Analysis\n{response}"
             current_summary_tokens += response_tokens
@@ -295,7 +298,8 @@ async def process_final_report(agent, file, chatbot_state: List[Dict[str, str]])
                             final_report_text += r.content
         except Exception as e:
             messages.append({"role": "assistant", "content": f"❌ Error generating final report: {str(e)}"})
-            return messages, report_path
+            yield messages, report_path
+            return
 
         final_report = f"# \U0001f9e0 Final Patient Report\n\n{clean_response(final_report_text)}"
         messages[-1]["content"] = f"📊 Final Report:\n\n{clean_response(final_report_text)}"
@@ -310,11 +314,13 @@ async def process_final_report(agent, file, chatbot_state: List[Dict[str, str]])
         messages.append({"role": "assistant", "content": f"✅ Report generated and saved: report_{timestamp}.md"})
         logger.info(f"Total processing time: {time.time() - start_time:.2f} seconds")
 
+        yield messages, report_path
+
     except Exception as e:
         messages.append({"role": "assistant", "content": f"❌ Error processing file: {str(e)}"})
         logger.error(f"Processing failed: {str(e)}")
-
-    return messages, report_path
+        yield messages, report_path
+        return
 
 async def create_ui(agent):
     """Create the Gradio UI for the patient history analysis tool."""