Edit file uploading

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 from main import (
     answer_query, 
     set_keys,
-    process_file
+    process_file,
+    handle_file
     )
 
 from pydantic import ConfigDict
@@ -18,7 +19,7 @@ setting_keys = gr.Interface(
 )
 
 uploading_files = gr.Interface(
-    fn=process_file,
+    fn=handle_file,
     inputs=gr.File(
         label="Upload a file",
         file_count="single",

main.py

@@ -1,8 +1,9 @@
 from utils.retriever import get_query_engine
 from utils.index import create_index
-from utils.constant import INDEX_PATH
+from utils.constant import INDEX_PATH, DATA_PATH
 import os
 from pathlib import Path
+import shutil
 
 def set_keys(co_api_key: str, llama_cloud_api_key: str) -> str:
     try:
@@ -30,7 +31,21 @@ def process_file(file) -> str:
         create_index(filepath, INDEX_PATH)
         return "File indexed successfully"
     except Exception as e:
-        return str(e)    
+        return str(e)   
+
+def handle_file(uploaded_file):
+    try:
+
+        temp_file_path = uploaded_file.name
+        save_file_path = os.path.join(DATA_PATH, "paper.pdf")
+        
+        shutil.move(temp_file_path, save_file_path)
+        
+        create_index(DATA_PATH, INDEX_PATH)
+        return "File uploaded and indexed"
+    
+    except Exception as e:
+        return str(e)
 
 def answer_query(query: str) -> str:
     

utils/constant.py

@@ -1,4 +1,4 @@
-DOC_PATH = "./data"
+DATA_PATH = "./data"
 INDEX_PATH = "./index"
 
 TOP_K_RETRIEVAL = 10  

utils/index.py

@@ -47,7 +47,7 @@ def load_index(path: str):
 
 if __name__ == "__main__":
 
-    doc_path = DOC_PATH
+    doc_path = DATA_PATH
     index_path = INDEX_PATH
 
     create_index(doc_path, index_path)