Update app.py

app.py

@@ -9,7 +9,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.chains.question_answering import load_qa_chain
 from langchain.prompts import PromptTemplate
 from dotenv import load_dotenv
-from google.api_core.exceptions import GoogleAPIError
+from google.api_core.exceptions import GoogleAPIError, InvalidArgument
 
 # Load environment variables
 load_dotenv()
@@ -39,8 +39,8 @@ def get_vector_store(chunks):
             model="models/embedding-001")  # type: ignore
         vector_store = FAISS.from_texts(chunks, embedding=embeddings)
         vector_store.save_local("faiss_index")
-    except GoogleAPIError:
-        st.error("Please try again in a minute.")
+    except (GoogleAPIError, InvalidArgument) as e:
+        st.error("Error processing embeddings. Please try again in a minute.")
 
 # Function to get conversational chain
 def get_conversational_chain():
@@ -60,8 +60,8 @@ def get_conversational_chain():
                                 input_variables=["context", "question"])
         chain = load_qa_chain(llm=model, chain_type="stuff", prompt=prompt)
         return chain
-    except GoogleAPIError:
-        st.error("Please try again in a minute.")
+    except (GoogleAPIError, InvalidArgument) as e:
+        st.error("Error creating conversational chain. Please try again in a minute.")
 
 # Function to clear chat history
 def clear_chat_history():
@@ -79,8 +79,8 @@ def user_input(user_question):
         response = chain(
             {"input_documents": docs, "question": user_question}, return_only_outputs=True)
         return response
-    except GoogleAPIError:
-        st.error("Please try again in a minute.")
+    except (GoogleAPIError, InvalidArgument) as e:
+        st.error("Error processing request. Please try again in a minute.")
 
 # Main function to run the Streamlit app
 def main():
@@ -178,5 +178,5 @@ def main():
                     else:
                         st.error("Please try again in a minute.")
 
-if __name__ == "main":
+if __name__ == "__main__":
     main()