Update app.py

app.py

@@ -126,9 +126,9 @@ def init_agent():
     return agent
 
 
-def stream_final_report(agent, file) -> Generator[List[Dict[str, str]], None, None]:
+def stream_final_report(agent, file) -> Generator[Tuple[List[Dict[str, str]], str], None, None]:
     if file is None or not hasattr(file, "name"):
-        yield [{"role": "assistant", "content": clean_response(stream_text)}]
+        yield ([{"role": "assistant", "content": "❌ Please upload a valid Excel file before analyzing."}], "")
         return
 
     extracted_text = extract_text_from_excel(file.name)
@@ -158,7 +158,7 @@ def stream_final_report(agent, file) -> Generator[List[Dict[str, str]], None, No
         chunk_responses.append(clean_response(response))
 
     final_prompt = "\n\n".join(chunk_responses) + "\n\nSummarize the key findings above."
-    yield [{"role": "user", "content": f"[Excel Uploaded: {file.name}]"}] 
+    messages = [{"role": "user", "content": f"[Excel Uploaded: {file.name}]"}]
     stream_text = ""
     for result in agent.run_gradio_chat(
         message=final_prompt,
@@ -177,14 +177,16 @@ def stream_final_report(agent, file) -> Generator[List[Dict[str, str]], None, No
             for r in result:
                 if hasattr(r, "content"):
                     stream_text += r.content
-        yield {"role": "assistant", "content": clean_response(stream_text)}
+        messages.append({"role": "assistant", "content": clean_response(stream_text)})
+        yield (messages, None)
 
     final_report = f"# \U0001f9e0 Final Patient Report\n\n{clean_response(stream_text)}"
     report_path = os.path.join(report_dir, f"report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md")
     with open(report_path, 'w') as f:
         f.write(final_report)
 
-    yield [{"role": "user", "content": f"[Excel Uploaded: {file.name}]"}, {"role": "assistant", "content": final_report}], report_path
+    messages.append({"role": "assistant", "content": final_report})
+    yield (messages, report_path)
 
 
 def create_ui(agent):
@@ -215,4 +217,4 @@ if __name__ == "__main__":
         )
     except Exception as e:
         print(f"Error: {str(e)}")
-        sys.exit(1)
+        sys.exit(1)