Update server.py
Browse files
server.py
CHANGED
@@ -535,36 +535,36 @@ async def run_workflow(request: WorkflowRequest):
|
|
535 |
|
536 |
agent_responses = {}
|
537 |
|
538 |
-
|
539 |
-
|
540 |
-
|
541 |
-
|
542 |
-
|
543 |
-
|
544 |
|
545 |
-
|
546 |
-
|
547 |
-
|
548 |
-
|
549 |
-
|
550 |
-
|
551 |
|
552 |
-
|
553 |
-
|
554 |
-
|
555 |
|
556 |
-
|
557 |
|
558 |
# Summary
|
559 |
-
|
560 |
-
|
561 |
-
|
562 |
-
|
563 |
-
|
564 |
-
|
565 |
-
|
566 |
-
|
567 |
-
|
568 |
|
569 |
# Generate CSV
|
570 |
csv_content = generate_csv_export(sample_alert, agent_responses)
|
|
|
535 |
|
536 |
agent_responses = {}
|
537 |
|
538 |
+
for agent_name, agent_func in agents:
|
539 |
+
workflow_results.extend([agent_name, "-" * 30])
|
540 |
+
try:
|
541 |
+
response = agent_func(sample_alert)
|
542 |
+
workflow_results.append("β
Message generated successfully")
|
543 |
+
agent_responses[agent_name] = response
|
544 |
|
545 |
+
# Add preview for some agents
|
546 |
+
if "WhatsApp" in agent_name:
|
547 |
+
text = response.get('text', 'N/A')
|
548 |
+
workflow_results.append(f" Preview: {text[:100]}..." if len(text) > 100 else f" Preview: {text}")
|
549 |
+
elif "SMS" in agent_name:
|
550 |
+
workflow_results.append(f" Preview: {str(response)[:100]}...")
|
551 |
|
552 |
+
except Exception as e:
|
553 |
+
workflow_results.append(f"β Error: {str(e)}")
|
554 |
+
agent_responses[agent_name] = f"Error: {str(e)}"
|
555 |
|
556 |
+
workflow_results.append("")
|
557 |
|
558 |
# Summary
|
559 |
+
workflow_results.extend([
|
560 |
+
"β
Workflow Summary",
|
561 |
+
"-" * 30,
|
562 |
+
f"π― Successfully generated alerts for {sample_alert['location']['village']}, {request.district}",
|
563 |
+
f"π Data Sources: {sample_alert['data_source']}",
|
564 |
+
f"π€ AI Enhanced: {'Yes' if sample_alert['alert']['ai_generated'] else 'No'}",
|
565 |
+
f"β° Generated at: {datetime.now().strftime('%Y-%m-%d %H:%M:%S UTC')}",
|
566 |
+
f"π± Agents Processed: {len([r for r in agent_responses.values() if not str(r).startswith('Error:')])}/{len(agents)}"
|
567 |
+
])
|
568 |
|
569 |
# Generate CSV
|
570 |
csv_content = generate_csv_export(sample_alert, agent_responses)
|