fruitpicker01 commited on
Commit
5862173
1 Parent(s): 290e351

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -3
app.py CHANGED
@@ -1,10 +1,21 @@
1
  import gradio as gr
 
 
2
 
3
  def process_input(uploaded_file, function):
4
  if uploaded_file is not None:
5
- # Предполагаем, что файл текстовый. Для других типов файлов потребуется соответствующая обработка.
6
- content = uploaded_file.read().decode("utf-8")
7
- # В этом месте можно добавить логику обработки содержимого файла
 
 
 
 
 
 
 
 
 
8
  return f"Результат для функции '{function}': {content[:100]}... (показаны первые 100 символов)"
9
  return "Файл не загружен или пустой"
10
 
 
1
  import gradio as gr
2
+ from io import BytesIO
3
+ import docx
4
 
5
  def process_input(uploaded_file, function):
6
  if uploaded_file is not None:
7
+ # Определяем тип файла по расширению
8
+ if uploaded_file.name.endswith('.txt'):
9
+ # Если файл текстовый, читаем как текст
10
+ content = uploaded_file.read().decode("utf-8")
11
+ elif uploaded_file.name.endswith('.docx'):
12
+ # Если файл Word, используем библиотеку python-docx
13
+ doc = docx.Document(BytesIO(uploaded_file.read()))
14
+ content = '\n'.join([para.text for para in doc.paragraphs])
15
+ else:
16
+ return "Неподдерживаемый тип файла"
17
+
18
+ # Тут можно добавить логику обработки текста
19
  return f"Результат для функции '{function}': {content[:100]}... (показаны первые 100 символов)"
20
  return "Файл не загружен или пустой"
21