fruitpicker01 commited on
Commit
3a3a439
·
verified ·
1 Parent(s): 6680111

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import gradio as gr
2
- import io
3
  import docx
4
 
5
  def process_input(uploaded_file, function):
@@ -7,16 +6,20 @@ def process_input(uploaded_file, function):
7
  return "Файл не загружен"
8
 
9
  try:
 
 
 
10
  # Читаем содержимое файла
11
- file_content = io.BytesIO(uploaded_file).read()
 
12
 
13
  # Определяем тип файла по расширению
14
- if uploaded_file.name.endswith('.txt'):
15
  # Если файл текстовый, декодируем содержимое
16
  content = file_content.decode("utf-8")
17
- elif uploaded_file.name.endswith('.docx'):
18
  # Если файл Word, используем библиотеку python-docx
19
- doc = docx.Document(io.BytesIO(file_content))
20
  content = '\n'.join([para.text for para in doc.paragraphs])
21
  else:
22
  return "Неподдерживаемый тип файла"
 
1
  import gradio as gr
 
2
  import docx
3
 
4
  def process_input(uploaded_file, function):
 
6
  return "Файл не загружен"
7
 
8
  try:
9
+ # Получаем путь к загруженному файлу
10
+ file_path = uploaded_file.name
11
+
12
  # Читаем содержимое файла
13
+ with open(file_path, "rb") as file:
14
+ file_content = file.read()
15
 
16
  # Определяем тип файла по расширению
17
+ if file_path.endswith('.txt'):
18
  # Если файл текстовый, декодируем содержимое
19
  content = file_content.decode("utf-8")
20
+ elif file_path.endswith('.docx'):
21
  # Если файл Word, используем библиотеку python-docx
22
+ doc = docx.Document(file_path)
23
  content = '\n'.join([para.text for para in doc.paragraphs])
24
  else:
25
  return "Неподдерживаемый тип файла"