Spaces:
Sleeping
Sleeping
RafalB commited on
Commit ·
379dfef
1
Parent(s): bdf26c9
Pass connected_services to /generate so ALEMBIC_API helper gets injected
Browse files
app.py
CHANGED
|
@@ -236,16 +236,19 @@ async def call_backend_agent(user_message: str, user_id: str, state: dict) -> di
|
|
| 236 |
return result
|
| 237 |
|
| 238 |
|
| 239 |
-
async def call_backend_generate(requirements: str, design_prefs: dict, user_id: str) -> dict:
|
| 240 |
if MOCK_MODE:
|
| 241 |
return await mock_build(requirements)
|
| 242 |
|
| 243 |
from model_runtime import _post
|
| 244 |
-
|
| 245 |
"requirements": requirements,
|
| 246 |
"design_prefs": design_prefs,
|
| 247 |
"user_id": user_id,
|
| 248 |
-
}
|
|
|
|
|
|
|
|
|
|
| 249 |
return result
|
| 250 |
|
| 251 |
|
|
@@ -763,7 +766,10 @@ async def handle_generate(
|
|
| 763 |
yield render_chat_html(session_state["chat_history"]), new_preview, gallery_html, session_state, "", HIDE_SAVE
|
| 764 |
|
| 765 |
try:
|
| 766 |
-
|
|
|
|
|
|
|
|
|
|
| 767 |
|
| 768 |
new_preview = BUILDING_HTML_TEMPLATE.format(
|
| 769 |
status="Deploying...", message="Uploading to CloudFront...", progress=85
|
|
@@ -875,7 +881,7 @@ async def handle_recheck_services(chat_html, preview_html, gallery_html, session
|
|
| 875 |
yield render_chat_html(session_state["chat_history"]), new_preview, gallery_html, session_state, "", HIDE_SAVE
|
| 876 |
|
| 877 |
try:
|
| 878 |
-
result = await call_backend_generate(requirements, {}, session_state["user_id"])
|
| 879 |
|
| 880 |
new_preview = BUILDING_HTML_TEMPLATE.format(
|
| 881 |
status="Deploying...", message="Uploading to CloudFront...", progress=85
|
|
|
|
| 236 |
return result
|
| 237 |
|
| 238 |
|
| 239 |
+
async def call_backend_generate(requirements: str, design_prefs: dict, user_id: str, connected_services: list = None) -> dict:
|
| 240 |
if MOCK_MODE:
|
| 241 |
return await mock_build(requirements)
|
| 242 |
|
| 243 |
from model_runtime import _post
|
| 244 |
+
payload = {
|
| 245 |
"requirements": requirements,
|
| 246 |
"design_prefs": design_prefs,
|
| 247 |
"user_id": user_id,
|
| 248 |
+
}
|
| 249 |
+
if connected_services:
|
| 250 |
+
payload["connected_services"] = connected_services
|
| 251 |
+
result = await _post("/generate", payload)
|
| 252 |
return result
|
| 253 |
|
| 254 |
|
|
|
|
| 766 |
yield render_chat_html(session_state["chat_history"]), new_preview, gallery_html, session_state, "", HIDE_SAVE
|
| 767 |
|
| 768 |
try:
|
| 769 |
+
# Pass connected services so codegen injects the API helper
|
| 770 |
+
services = (session_state.get("pending_build", {}).get("required_services")
|
| 771 |
+
or all_required_services or [])
|
| 772 |
+
result = await call_backend_generate(requirements, design_prefs, session_state["user_id"], services)
|
| 773 |
|
| 774 |
new_preview = BUILDING_HTML_TEMPLATE.format(
|
| 775 |
status="Deploying...", message="Uploading to CloudFront...", progress=85
|
|
|
|
| 881 |
yield render_chat_html(session_state["chat_history"]), new_preview, gallery_html, session_state, "", HIDE_SAVE
|
| 882 |
|
| 883 |
try:
|
| 884 |
+
result = await call_backend_generate(requirements, {}, session_state["user_id"], required_services)
|
| 885 |
|
| 886 |
new_preview = BUILDING_HTML_TEMPLATE.format(
|
| 887 |
status="Deploying...", message="Uploading to CloudFront...", progress=85
|