Update error handling
Browse files
app.py
CHANGED
|
@@ -13,19 +13,22 @@ retriever = Retrieval(model_name=embedding_model_name)
|
|
| 13 |
|
| 14 |
llm_model_name = "gpt-4o-mini"
|
| 15 |
# Settting up LLMGenerator
|
| 16 |
-
llm_generator =
|
| 17 |
|
| 18 |
-
def set_api_key(api_key):
|
| 19 |
-
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
def process_inputs(api_key:str, pdf_file, questions: str):
|
| 22 |
|
| 23 |
# Setup Api KEY
|
| 24 |
set_api_key(api_key)
|
|
|
|
| 25 |
|
| 26 |
if pdf_file is None:
|
| 27 |
-
raise
|
| 28 |
-
|
| 29 |
|
| 30 |
# Parsing the pdf
|
| 31 |
doc_handler = DocParsing(file_path=pdf_file.name,model_name=embedding_model_name)
|
|
@@ -34,7 +37,11 @@ def process_inputs(api_key:str, pdf_file, questions: str):
|
|
| 34 |
# Create vector store
|
| 35 |
retriever.create_vector_store(chunks=docs)
|
| 36 |
|
|
|
|
|
|
|
| 37 |
|
|
|
|
|
|
|
| 38 |
output_dict = {}
|
| 39 |
questions_list = questions.strip().split('\n')
|
| 40 |
for question in questions_list:
|
|
|
|
| 13 |
|
| 14 |
llm_model_name = "gpt-4o-mini"
|
| 15 |
# Settting up LLMGenerator
|
| 16 |
+
llm_generator = None
|
| 17 |
|
| 18 |
+
def set_api_key(api_key: str):
|
| 19 |
+
if api_key.strip():
|
| 20 |
+
os.environ['OPENAI_API_KEY'] = api_key
|
| 21 |
+
else:
|
| 22 |
+
raise gr.Error("Please provide a valid API key")
|
| 23 |
|
| 24 |
def process_inputs(api_key:str, pdf_file, questions: str):
|
| 25 |
|
| 26 |
# Setup Api KEY
|
| 27 |
set_api_key(api_key)
|
| 28 |
+
|
| 29 |
|
| 30 |
if pdf_file is None:
|
| 31 |
+
raise gr.Error("Please upload a pdf file")
|
|
|
|
| 32 |
|
| 33 |
# Parsing the pdf
|
| 34 |
doc_handler = DocParsing(file_path=pdf_file.name,model_name=embedding_model_name)
|
|
|
|
| 37 |
# Create vector store
|
| 38 |
retriever.create_vector_store(chunks=docs)
|
| 39 |
|
| 40 |
+
# LLM Generator
|
| 41 |
+
llm_generator = LLMGeneration(llm_model_name=llm_model_name)
|
| 42 |
|
| 43 |
+
if not questions.strip():
|
| 44 |
+
raise gr.Error("Please provide valid set of questions")
|
| 45 |
output_dict = {}
|
| 46 |
questions_list = questions.strip().split('\n')
|
| 47 |
for question in questions_list:
|