Update main.py

main.py

@@ -94,3 +94,47 @@ async def predict(data: Predict):
         return prompt_response_dict
     else:
         raise HTTPException(status_code=400, detail="Prompt Incorrect")
+
+@app.get("/run_ingest")
+def run_ingest_route():
+    try:
+        if os.path.exists(PERSIST_DIRECTORY):
+            try:
+                shutil.rmtree(PERSIST_DIRECTORY)
+            except OSError as e:
+                print(f"Error: {e.filename} - {e.strerror}.")
+        else:
+            print("The directory does not exist")
+
+        run_langest_commands = ["python", "ingest.py"]
+        if DEVICE_TYPE == "cpu":
+            run_langest_commands.append("--device_type")
+            run_langest_commands.append(DEVICE_TYPE)
+
+        result = subprocess.run(run_langest_commands, capture_output=True)
+        
+        if result.returncode != 0:
+            raise HTTPException(status_code=400, detail="Script execution failed: {}")
+            
+        # load the vectorstore
+        DB = Chroma(
+            persist_directory=PERSIST_DIRECTORY,
+            embedding_function=EMBEDDINGS,
+            client_settings=CHROMA_SETTINGS,
+        )
+        RETRIEVER = DB.as_retriever()
+        prompt, memory = get_prompt_template(promptTemplate_type="llama", history=False)
+
+        QA = RetrievalQA.from_chain_type(
+            llm=LLM,
+            chain_type="stuff",
+            retriever=RETRIEVER,
+            return_source_documents=SHOW_SOURCES,
+            chain_type_kwargs={
+                "prompt": prompt,
+            },
+        )
+        return "Script executed successfully: {}".format(result.stdout.decode("utf-8"))
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error occurred: {str(e)}")
+