RAG-retrieval / app.py
yasirme's picture
1
9772c46
raw
history blame
1.2 kB
import os
from flask import Flask, send_file, request, jsonify
from utils.handle_file import file_handler
from werkzeug.utils import secure_filename
app = Flask(__name__)
app.config['MAX_CONTENT_LENGTH'] = 11 * 1024 * 1024
@app.route("/")
def index():
return send_file('src/index.html')
@app.route('/upload', methods=['POST'])
def upload():
try:
allowed_chars = request.args.get('allowed_size')
if 'file' not in request.files and 'files' not in request.files:
return jsonify({"error": "No files uploaded"}), 400
if 'files' in request.files:
files = request.files.getlist('files')
else:
files = request.files.getlist('file')
if not files or not files[0].filename:
return jsonify({"error": "No files selected"}), 400
if len(files) == 1:
return file_handler.process_file(files[0], allowed_chars)
else:
return file_handler.process_files(files, allowed_chars)
except Exception as e:
return jsonify({"error": f"An error occurred: {e}"}), 500
def main():
app.run(host='0.0.0.0', port=7860, debug=True)
if __name__ == "__main__":
main()