Spaces:
Running
Running
File size: 1,063 Bytes
44870e3 24055cd 44870e3 54155f3 44870e3 54155f3 9772c46 54155f3 44870e3 54155f3 44870e3 24055cd 54155f3 24055cd 44870e3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import os
from flask import Flask, send_file, request, jsonify
from utils.handle_file import file_handler
from werkzeug.utils import secure_filename
from rag.RAG import rag
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_chars")
if 'files' not in request.files:
return jsonify({"error": "No files uploaded"}), 400
files = request.files.getlist('files')
return file_handler.handle_files(files=files,allowed_chars=allowed_chars)
except Exception as e:
return jsonify({"error": f"An error occurred: {e} "}), 500
@app.route('/embedding', methods=['POST'])
def embedding():
return rag.generate_embedding(
text=request.json.get("text"),
task_type=request.json.get("task_type")
)
def main():
app.run(host='0.0.0.0', port=7860, debug=True)
if __name__ == "__main__":
main()
|