1

app.py

@@ -2,6 +2,8 @@ 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
 
@@ -14,6 +16,7 @@ def index():
 def upload():
     try:
         allowed_chars = request.args.get('allowed_size')
+        print(allowed_chars)
         if 'file' not in request.files and 'files' not in request.files:
             return jsonify({"error": "No files uploaded"}), 400
         if 'files' in request.files:
@@ -30,6 +33,14 @@ def upload():
     except Exception as e:
         return jsonify({"error": f"An error occurred: {e}"}), 500
 
+
+@app.route('/embedding')
+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)
 

utils/handle_file.py

@@ -115,7 +115,8 @@ class FileHandler:
         try:
             allowed_limit = self._validate_params(allowed_chars)
             content = self.read_file(file)
-            if allowed_limit and len(content) > allowed_limit:
+
+            if len(content) > allowed_limit:
                 return {"error": f"Character count ({len(content)}) exceeds the allowed limit ({allowed_limit})"}, 400                
             return rag.generate_embedding(content)
             
@@ -136,7 +137,7 @@ class FileHandler:
                     total_chars += len(content)
                 except ValueError as e:
                     return {"error": f"Error with file '{file.filename}': {str(e)}"}, 400            
-            if allowed_limit and total_chars > allowed_limit:
+            if total_chars > allowed_limit:
                 return {"error": f"Total character count ({total_chars}) exceeds the allowed limit ({allowed_limit})"}, 400
                                 
             results = []