soiz1 commited on
Commit
1f09515
Β·
verified Β·
1 Parent(s): 464d134

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -16
app.py CHANGED
@@ -2,27 +2,29 @@ import gradio as gr
2
  import requests
3
  from io import BytesIO
4
  from zipfile import ZipFile
 
 
5
 
6
  def urls_to_zip(urls_text):
7
  urls = urls_text.strip().split("\n")
8
- zip_buffer = BytesIO()
9
 
10
- with ZipFile(zip_buffer, "w") as zip_file:
11
- for i, url in enumerate(urls):
12
- url = url.strip()
13
- if not url:
14
- continue
15
- try:
16
- response = requests.get(url)
17
- response.raise_for_status()
18
- filename = url.split("/")[-1] or f"file_{i}"
19
- zip_file.writestr(filename, response.content)
20
- except Exception as e:
21
- # γ‚¨γƒ©γƒΌγŒγ‚γ‚Œγ°γƒ•γ‚‘γ‚€γƒ«εγ«γ‚¨γƒ©γƒΌζƒ…ε ±γ‚’ε…₯γ‚Œγ‚‹οΌˆδ»»ζ„οΌ‰
22
- zip_file.writestr(f"error_{i}.txt", f"Failed to download {url}\nError: {str(e)}")
 
 
23
 
24
- zip_buffer.seek(0)
25
- return zip_buffer
26
 
27
  with gr.Blocks() as demo:
28
  gr.Markdown("### ζ”Ήθ‘Œγ§εŒΊεˆ‡γ£γŸURLγƒͺγ‚Ήγƒˆγ‹γ‚‰ZIPγ‚’δ½œζˆγ—γΎγ™")
 
2
  import requests
3
  from io import BytesIO
4
  from zipfile import ZipFile
5
+ import tempfile
6
+ import os
7
 
8
  def urls_to_zip(urls_text):
9
  urls = urls_text.strip().split("\n")
 
10
 
11
+ # δΈ€ζ™‚γƒ•γ‚‘γ‚€γƒ«γ‚’δ½œζˆ
12
+ with tempfile.NamedTemporaryFile(delete=False, suffix=".zip") as tmp:
13
+ with ZipFile(tmp, "w") as zip_file:
14
+ for i, url in enumerate(urls):
15
+ url = url.strip()
16
+ if not url:
17
+ continue
18
+ try:
19
+ response = requests.get(url)
20
+ response.raise_for_status()
21
+ filename = url.split("/")[-1] or f"file_{i}"
22
+ zip_file.writestr(filename, response.content)
23
+ except Exception as e:
24
+ zip_file.writestr(f"error_{i}.txt", f"Failed to download {url}\nError: {str(e)}")
25
+ tmp_path = tmp.name
26
 
27
+ return tmp_path # フゑむルパスを返す
 
28
 
29
  with gr.Blocks() as demo:
30
  gr.Markdown("### ζ”Ήθ‘Œγ§εŒΊεˆ‡γ£γŸURLγƒͺγ‚Ήγƒˆγ‹γ‚‰ZIPγ‚’δ½œζˆγ—γΎγ™")