Spaces:
Running
Running
add sdk options
Browse files
app.py
CHANGED
|
@@ -1166,6 +1166,18 @@ with gr.Blocks(
|
|
| 1166 |
lines=1,
|
| 1167 |
visible=True
|
| 1168 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1169 |
deploy_btn = gr.Button("π Deploy App", variant="primary")
|
| 1170 |
deploy_status = gr.Markdown(visible=False, label="Deploy status")
|
| 1171 |
input = gr.Textbox(
|
|
@@ -1286,37 +1298,47 @@ with gr.Blocks(
|
|
| 1286 |
def deploy_to_user_space(
|
| 1287 |
code,
|
| 1288 |
space_name,
|
|
|
|
| 1289 |
profile: gr.OAuthProfile | None = None,
|
| 1290 |
token: gr.OAuthToken | None = None
|
| 1291 |
):
|
| 1292 |
if not code or not code.strip():
|
| 1293 |
return gr.update(value="No code to deploy.", visible=True)
|
| 1294 |
if profile is None or token is None:
|
| 1295 |
-
# Fallback to old method if not logged in
|
| 1296 |
return gr.update(value="Please log in with your Hugging Face account to deploy to your own Space. Otherwise, use the default deploy (opens in new tab).", visible=True)
|
| 1297 |
username = profile.username
|
| 1298 |
repo_id = f"{username}/{space_name.strip()}"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1299 |
api = HfApi(token=token.token)
|
| 1300 |
# Create the Space if it doesn't exist
|
| 1301 |
try:
|
| 1302 |
-
# Fix create_repo call: use repo_id, not name
|
| 1303 |
api.create_repo(
|
| 1304 |
repo_id=repo_id, # e.g. username/space_name
|
| 1305 |
repo_type="space",
|
| 1306 |
-
space_sdk=
|
| 1307 |
exist_ok=True # Don't error if it already exists
|
| 1308 |
)
|
| 1309 |
except Exception as e:
|
| 1310 |
return gr.update(value=f"Error creating Space: {e}", visible=True)
|
| 1311 |
# Save code to a temporary file
|
| 1312 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1313 |
f.write(code)
|
| 1314 |
temp_path = f.name
|
| 1315 |
# Upload the file
|
| 1316 |
try:
|
| 1317 |
api.upload_file(
|
| 1318 |
path_or_fileobj=temp_path,
|
| 1319 |
-
path_in_repo=
|
| 1320 |
repo_id=repo_id,
|
| 1321 |
repo_type="space"
|
| 1322 |
)
|
|
@@ -1328,7 +1350,7 @@ with gr.Blocks(
|
|
| 1328 |
# Connect the deploy button to the new function
|
| 1329 |
deploy_btn.click(
|
| 1330 |
deploy_to_user_space,
|
| 1331 |
-
inputs=[code_output, space_name_input],
|
| 1332 |
outputs=deploy_status
|
| 1333 |
)
|
| 1334 |
# Keep the old deploy method as fallback (if not logged in, user can still use the old method)
|
|
|
|
| 1166 |
lines=1,
|
| 1167 |
visible=True
|
| 1168 |
)
|
| 1169 |
+
# Add SDK selection dropdown
|
| 1170 |
+
sdk_choices = [
|
| 1171 |
+
("Gradio (Python)", "gradio"),
|
| 1172 |
+
("Streamlit (Python)", "streamlit"),
|
| 1173 |
+
("Static (HTML)", "static")
|
| 1174 |
+
]
|
| 1175 |
+
sdk_dropdown = gr.Dropdown(
|
| 1176 |
+
choices=[x[0] for x in sdk_choices],
|
| 1177 |
+
value="Gradio (Python)",
|
| 1178 |
+
label="App SDK",
|
| 1179 |
+
visible=True
|
| 1180 |
+
)
|
| 1181 |
deploy_btn = gr.Button("π Deploy App", variant="primary")
|
| 1182 |
deploy_status = gr.Markdown(visible=False, label="Deploy status")
|
| 1183 |
input = gr.Textbox(
|
|
|
|
| 1298 |
def deploy_to_user_space(
|
| 1299 |
code,
|
| 1300 |
space_name,
|
| 1301 |
+
sdk_name, # new argument
|
| 1302 |
profile: gr.OAuthProfile | None = None,
|
| 1303 |
token: gr.OAuthToken | None = None
|
| 1304 |
):
|
| 1305 |
if not code or not code.strip():
|
| 1306 |
return gr.update(value="No code to deploy.", visible=True)
|
| 1307 |
if profile is None or token is None:
|
|
|
|
| 1308 |
return gr.update(value="Please log in with your Hugging Face account to deploy to your own Space. Otherwise, use the default deploy (opens in new tab).", visible=True)
|
| 1309 |
username = profile.username
|
| 1310 |
repo_id = f"{username}/{space_name.strip()}"
|
| 1311 |
+
# Map SDK name to HF SDK slug
|
| 1312 |
+
sdk_map = {
|
| 1313 |
+
"Gradio (Python)": "gradio",
|
| 1314 |
+
"Streamlit (Python)": "streamlit",
|
| 1315 |
+
"Static (HTML)": "static"
|
| 1316 |
+
}
|
| 1317 |
+
sdk = sdk_map.get(sdk_name, "gradio")
|
| 1318 |
api = HfApi(token=token.token)
|
| 1319 |
# Create the Space if it doesn't exist
|
| 1320 |
try:
|
|
|
|
| 1321 |
api.create_repo(
|
| 1322 |
repo_id=repo_id, # e.g. username/space_name
|
| 1323 |
repo_type="space",
|
| 1324 |
+
space_sdk=sdk, # Use selected SDK
|
| 1325 |
exist_ok=True # Don't error if it already exists
|
| 1326 |
)
|
| 1327 |
except Exception as e:
|
| 1328 |
return gr.update(value=f"Error creating Space: {e}", visible=True)
|
| 1329 |
# Save code to a temporary file
|
| 1330 |
+
if sdk == "static":
|
| 1331 |
+
file_name = "index.html"
|
| 1332 |
+
else:
|
| 1333 |
+
file_name = "app.py"
|
| 1334 |
+
with tempfile.NamedTemporaryFile("w", suffix=f".{file_name.split('.')[-1]}", delete=False) as f:
|
| 1335 |
f.write(code)
|
| 1336 |
temp_path = f.name
|
| 1337 |
# Upload the file
|
| 1338 |
try:
|
| 1339 |
api.upload_file(
|
| 1340 |
path_or_fileobj=temp_path,
|
| 1341 |
+
path_in_repo=file_name,
|
| 1342 |
repo_id=repo_id,
|
| 1343 |
repo_type="space"
|
| 1344 |
)
|
|
|
|
| 1350 |
# Connect the deploy button to the new function
|
| 1351 |
deploy_btn.click(
|
| 1352 |
deploy_to_user_space,
|
| 1353 |
+
inputs=[code_output, space_name_input, sdk_dropdown],
|
| 1354 |
outputs=deploy_status
|
| 1355 |
)
|
| 1356 |
# Keep the old deploy method as fallback (if not logged in, user can still use the old method)
|