Update main.py

main.py

@@ -91,7 +91,7 @@ async def predict(data: Predict):
                 (os.path.basename(str(document.metadata["source"])), str(document.page_content))
             )
 
-        return {response: prompt_response_dict}
+        return {"response": prompt_response_dict}
     else:
         raise HTTPException(status_code=400, detail="Prompt Incorrect")
 
@@ -102,9 +102,9 @@ def run_ingest_route():
             try:
                 shutil.rmtree(PERSIST_DIRECTORY)
             except OSError as e:
-                print(f"Error: {e.filename} - {e.strerror}.")
+                raise HTTPException(status_code=500, detail=f"Error: {e.filename} - {e.strerror}.")
         else:
-            print("The directory does not exist")
+            raise HTTPException(status_code=500, detail="The directory does not exist")
 
         run_langest_commands = ["python", "ingest.py"]
         if DEVICE_TYPE == "cpu":
@@ -134,7 +134,9 @@ def run_ingest_route():
                 "prompt": prompt,
             },
         )
-        return "Script executed successfully: {}".format(result.stdout.decode("utf-8"))
+
+        response = "Script executed successfully: {}".format(result.stdout.decode("utf-8"))
+        return {"response": response}
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error occurred: {str(e)}")