Update app.py

app.py

@@ -1,20 +1,11 @@
 import gradio as gr
-#import pandas as pd
 import logging, os, sys, threading
 
-#from dotenv import load_dotenv, find_dotenv
 from custom_utils import connect_to_database, rag_ingestion, rag_retrieval, rag_inference
 
-#from pydantic import BaseModel
-#from typing import Optional
-
-#from IPython.display import display, HTML
-
 lock = threading.Lock()
 
-#_ = load_dotenv(find_dotenv())
-
-RAG_INGESTION = True
+RAG_INGESTION = False
 
 RAG_OFF      = "Off"
 RAG_NAIVE    = "Naive RAG"
@@ -35,8 +26,7 @@ def invoke(openai_api_key, prompt, rag_option):
         db, collection = connect_to_database()
 
         if (RAG_INGESTION):
-            rag_ingestion(collection)
-            return "Done"
+            return rag_ingestion(collection)
         else:
             """
             match_stage = {
@@ -50,8 +40,7 @@ def invoke(openai_api_key, prompt, rag_option):
             """
     
             search_results = rag_retrieval(openai_api_key, prompt, db, collection)
-            result  = rag_inference(openai_api_key, prompt, search_results)
-            return result
+            return rag_inference(openai_api_key, prompt, search_results)
 
 gr.close_all()