Update QA_Bot.py

QA_Bot.py

@@ -1,11 +1,14 @@
 import streamlit as st
 from QnA import Q_A
 import re,time
-from QnA import get_hugging_face_model  , summarize
+from QnA import get_hugging_face_model  , summarize ,get_groq_model
 
 
-def summarize_data(documents):
-    llm  = get_hugging_face_model
+def summarize_data(documents,api_key):
+    if api_key.startswith('gsk'):
+        llm  = get_groq_model(api_key)
+    else:
+        llm= get_hugging_face_model(api_key=api_key)
     summary = summarize(documents,llm)
     return summary
     
@@ -21,7 +24,7 @@ def QA_Bot(vectorstore,API_KEY,documents):
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
-    summary_response = summarize_data(documents)
+    summary_response = summarize_data(documents,API_KEY)
     print(summary_response)
     
     # React to user input