Update app.py

app.py

@@ -138,7 +138,7 @@ def stream_final_report(agent, file) -> Generator[Tuple[List[Dict[str, str]], Un
 
     try:
         # Initial processing message
-        messages = [{"role": "user", "content": f"Processing Excel file: {file.name}"},
+        messages = [{"role": "user", "content": f"Processing Excel file: {os.path.basename(file.name)}"},
                    {"role": "assistant", "content": "⏳ Extracting and analyzing data..."}]
         yield messages, None
 
@@ -214,7 +214,7 @@ def stream_final_report(agent, file) -> Generator[Tuple[List[Dict[str, str]], Un
         yield messages, report_path
 
     except Exception as e:
-        messages.append({"role": "assistant", "content": f"❌ Error: {str(e)}"})
+        messages.append({"role": "assistant", "content": f"❌ Error processing file: {str(e)}"})
         yield messages, None
 
 
@@ -228,6 +228,7 @@ def create_ui(agent):
                     label="Clinical Assistant",
                     show_copy_button=True,
                     height=600,
+                    type="messages",
                     avatar_images=(
                         None,  # User avatar
                         "https://i.imgur.com/6wX7Zb4.png"  # Bot avatar
@@ -248,11 +249,6 @@ def create_ui(agent):
                     visible=False,
                     interactive=False
                 )
-                gr.Examples(
-                    examples=["sample_data/sample_patient_history.xlsx"],
-                    inputs=file_upload,
-                    label="Sample Files"
-                )
 
         analyze_btn.click(
             fn=lambda file: stream_final_report(agent, file),
@@ -266,11 +262,6 @@ def create_ui(agent):
                 return gr.File(visible=True, value=report_path)
             return gr.File(visible=False)
 
-        demo.load(
-            lambda: None,
-            outputs=report_output
-        )
-
     return demo