pippobertin commited on
Commit
792112a
·
verified ·
1 Parent(s): 71ba61a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -9
app.py CHANGED
@@ -3,29 +3,37 @@ import os
3
  import zipfile
4
  from spacy2 import classifica_e_sposta_documenti
5
 
6
- def classifica_e_sposta_documenti_wrapper(upload):
7
- folder_path = upload.name
 
 
 
 
 
 
 
8
  # Chiamata alla tua funzione spacy2.classifica_e_sposta_documenti
9
- classifica_e_sposta_documenti(folder_path)
10
 
11
  # Creazione del file zip
12
  zip_filename = 'processed_files.zip'
13
- zip_filepath = os.path.join(folder_path, zip_filename)
14
  with zipfile.ZipFile(zip_filepath, 'w', zipfile.ZIP_DEFLATED) as zipf:
15
- for root, dirs, files in os.walk(folder_path):
16
  for file in files:
17
  zipf.write(os.path.join(root, file),
18
  os.path.relpath(os.path.join(root, file),
19
- os.path.join(folder_path, '..')))
20
 
21
  return zip_filepath
22
 
23
- def process_files(upload):
24
  # Chiamata alla tua funzione spacy2.classifica_e_sposta_documenti
25
- zip_filepath = classifica_e_sposta_documenti_wrapper(upload)
26
 
27
  return "Elaborazione completata. Puoi scaricare i file elaborati [qui](" + zip_filepath + ")."
28
 
29
- iface = gr.Interface(fn=process_files, inputs="file", outputs="text", live=True)
30
  iface.launch()
31
 
 
 
3
  import zipfile
4
  from spacy2 import classifica_e_sposta_documenti
5
 
6
+ def classifica_e_sposta_documenti_wrapper(uploaded_files):
7
+ # Crea una cartella temporanea per archiviare i file caricati
8
+ temp_folder = gr.utils.get_file_directory(uploaded_files[0].name)
9
+
10
+ # Salva tutti i file caricati nella cartella temporanea
11
+ for uploaded_file in uploaded_files:
12
+ file_path = os.path.join(temp_folder, uploaded_file.name)
13
+ uploaded_file.download(file_path)
14
+
15
  # Chiamata alla tua funzione spacy2.classifica_e_sposta_documenti
16
+ classifica_e_sposta_documenti(temp_folder)
17
 
18
  # Creazione del file zip
19
  zip_filename = 'processed_files.zip'
20
+ zip_filepath = os.path.join(temp_folder, zip_filename)
21
  with zipfile.ZipFile(zip_filepath, 'w', zipfile.ZIP_DEFLATED) as zipf:
22
+ for root, dirs, files in os.walk(temp_folder):
23
  for file in files:
24
  zipf.write(os.path.join(root, file),
25
  os.path.relpath(os.path.join(root, file),
26
+ os.path.join(temp_folder, '..')))
27
 
28
  return zip_filepath
29
 
30
+ def process_files(uploaded_files):
31
  # Chiamata alla tua funzione spacy2.classifica_e_sposta_documenti
32
+ zip_filepath = classifica_e_sposta_documenti_wrapper(uploaded_files)
33
 
34
  return "Elaborazione completata. Puoi scaricare i file elaborati [qui](" + zip_filepath + ")."
35
 
36
+ iface = gr.Interface(fn=process_files, inputs="files", outputs="text", live=True)
37
  iface.launch()
38
 
39
+