Update app.py

app.py

@@ -3,7 +3,10 @@ from io import BytesIO
 import docx
 
 def process_input(uploaded_file, function):
-    if uploaded_file is not None:
+    if uploaded_file is None:
+        return "Файл не загружен"
+
+    try:
         # Определяем тип файла по расширению
         if uploaded_file.name.endswith('.txt'):
             # Если файл текстовый, читаем как текст
@@ -14,16 +17,17 @@ def process_input(uploaded_file, function):
             content = '\n'.join([para.text for para in doc.paragraphs])
         else:
             return "Неподдерживаемый тип файла"
-        
+
         # Тут можно добавить логику обработки текста
         return f"Результат для функции '{function}': {content[:100]}... (показаны первые 100 символов)"
-    return "Файл не загружен или пустой"
+    except Exception as e:
+        return f"Ошибка при обработке файла: {str(e)}"
 
 def main():
     with gr.Blocks() as demo:
         gr.Markdown("### AI Research Assistant")
         with gr.Row():
-            file_input = gr.File(label="Загрузите файл")
+            file_input = gr.File(label="Загрузите файл", file_types=["text", "docx"])
             function_select = gr.Dropdown(choices=[
                 "Суммаризатор", "Поиск новых статей", "Учитель", "Критик", "Тестировщик",
                 "Визуализатор связей", "Советник", "Соавтор", "Переводчик", "Аннотатор",
@@ -39,7 +43,6 @@ def main():
             inputs=[file_input, function_select],
             outputs=output_text
         )
-
     demo.launch()
 
 if __name__ == "__main__":