Update app.py
Browse files
app.py
CHANGED
|
@@ -198,10 +198,11 @@ async def process_file(agent: TxAgent, file_path: str) -> Generator[Tuple[List[D
|
|
| 198 |
with ThreadPoolExecutor(max_workers=MAX_CONCURRENT) as executor:
|
| 199 |
futures = []
|
| 200 |
for idx, chunk in enumerate(chunks):
|
| 201 |
-
|
| 202 |
lambda c, i: asyncio.run(process_chunk(agent, c, i)),
|
| 203 |
chunk, idx
|
| 204 |
)
|
|
|
|
| 205 |
messages.append({"role": "assistant", "content": f"π Processing chunk {idx+1}/{len(chunks)}..."})
|
| 206 |
yield messages, None
|
| 207 |
|
|
|
|
| 198 |
with ThreadPoolExecutor(max_workers=MAX_CONCURRENT) as executor:
|
| 199 |
futures = []
|
| 200 |
for idx, chunk in enumerate(chunks):
|
| 201 |
+
future = executor.submit(
|
| 202 |
lambda c, i: asyncio.run(process_chunk(agent, c, i)),
|
| 203 |
chunk, idx
|
| 204 |
)
|
| 205 |
+
futures.append(future)
|
| 206 |
messages.append({"role": "assistant", "content": f"π Processing chunk {idx+1}/{len(chunks)}..."})
|
| 207 |
yield messages, None
|
| 208 |
|