Update app.py

app.py

@@ -109,6 +109,11 @@ def upload_file(file, model_name, split_strategy, chunk_size, overlap_size, max_
     #    f.write(file.read())  # Read and write the file content
 
     # Process files and get embeddings
+    try:
+        chunk_size = int(chunk_size)  # Convert chunk_size to int
+        overlap_size = int(overlap_size)  # Convert overlap_size to int
+    except ValueError:
+        return {"error": "Chunk size and overlap size must be valid integers."}
 
     if overlap_size is None:
         overlap_size = 0  # Set a default overlap size
@@ -134,10 +139,10 @@ iface = gr.Interface(
         gr.Textbox(label="Search Query"),
         gr.Dropdown(choices=list(MODELS.keys()), label="Embedding Model"),
         gr.Radio(choices=["sentence", "recursive"], label="Split Strategy"),
-        gr.Slider(100, 1000, step=100, value=500, label="Chunk Size", type='int'),  # Ensure type is int
-        gr.Slider(0, 100, step=10, value=50, label="Overlap Size", type='int'),  # Ensure type is int
-        gr.Slider(50, 500, step=50, value=200, label="Max Tokens", type='int'),  # Ensure type is int
-        gr.Slider(1, 10, step=1, value=5, label="Top K", type='int')  # Ensure type is int
+        gr.Slider(100, 1000, step=100, value=500, label="Chunk Size"),  # Ensure type is int
+        gr.Slider(0, 100, step=10, value=50, label="Overlap Size),  # Ensure type is int
+        gr.Slider(50, 500, step=50, value=200, label="Max Tokens"),  # Ensure type is int
+        gr.Slider(1, 10, step=1, value=5, label="Top K")  # Ensure type is int
     ],
     outputs="json"
 )