Update app.py

app.py

@@ -191,14 +191,13 @@ async def process_chunk(agent, chunk: str, chunk_index: int, total_chunks: int)
     return chunk_index, clean_response(response), status
 
 async def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tuple[List[Dict[str, str]], Union[str, None]]:
-    """Process the Excel file and generate a final report with asynchronous updates."""
+    """Process the Excel file and generate a final report."""
     messages = chatbot_state if chatbot_state else []
     report_path = None
 
     if file is None or not hasattr(file, "name"):
         messages.append({"role": "assistant", "content": "❌ Please upload a valid Excel file before analyzing."})
-        yield messages, report_path
-        return
+        return messages, report_path
 
     try:
         messages.append({"role": "user", "content": f"Processing Excel file: {os.path.basename(file.name)}"})
@@ -228,14 +227,12 @@ async def process_final_report(agent, file, chatbot_state: List[Dict[str, str]])
                     chunk_index, response, status = future.result()
                     chunk_responses[chunk_index] = response
                     messages.append({"role": "assistant", "content": status})
-                    yield messages, None
 
         # Filter out empty responses
         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."})
-            yield messages, report_path
-            return
+            return messages, report_path
 
         # Summarize chunk responses incrementally
         summary = ""
@@ -267,8 +264,7 @@ 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)}"})
-                    yield messages, report_path
-                    return
+                    return messages, report_path
 
             summary += f"\n\n### Chunk {i+1} Analysis\n{response}"
             current_summary_tokens += response_tokens
@@ -295,11 +291,10 @@ async def process_final_report(agent, file, chatbot_state: List[Dict[str, str]])
                 elif isinstance(result, list):
                     for r in result:
                         if hasattr(r, "content"):
-                            final_report_text += r.content
+                            summary_response += r.content
         except Exception as e:
             messages.append({"role": "assistant", "content": f"❌ Error generating final report: {str(e)}"})
-            yield messages, report_path
-            return
+            return messages, report_path
 
         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)}"
@@ -314,13 +309,12 @@ 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
+        return 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)}")
-        yield messages, report_path
-        return
+        return messages, report_path
 
 async def create_ui(agent):
     """Create the Gradio UI for the patient history analysis tool."""
@@ -360,13 +354,9 @@ async def create_ui(agent):
 
         async def update_ui(file, current_state):
             messages = current_state if current_state else []
-            report_path = None
-            async for new_messages, new_report_path in process_final_report(agent, file, messages):
-                messages = new_messages
-                report_path = new_report_path
-                report_update = gr.update(visible=report_path is not None, value=report_path)
-                yield messages, report_update, messages
-            yield messages, gr.update(visible=report_path is not None, value=report_path), messages
+            messages, report_path = await process_final_report(agent, file, messages)
+            report_update = gr.update(visible=report_path is not None, value=report_path)
+            return messages, report_update, messages
 
         analyze_btn.click(
             fn=update_ui,
@@ -386,7 +376,9 @@ if __name__ == "__main__":
             server_port=7860,
             show_error=True,
             allowed_paths=["/data/hf_cache/reports"],
-            share=False
+            share=False,
+            inline=False,
+            max_threads=40
         )
     except Exception as e:
         print(f"Error: {str(e)}")