Update main.py

main.py

@@ -23,7 +23,7 @@ DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
 GOOGLE_API_KEY = os.environ['GOOGLE_API_KEY']
 
 genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel('gemini-pro')
+gemini_model = genai.GenerativeModel('gemini-pro')
 
 
 embeddings = HuggingFaceInstructEmbeddings(
@@ -174,11 +174,12 @@ async def llm_response(chain,id,mode):
 
         try:
             print("\n google gemini===================\n")
-            chat = model.start_chat(history=[])
+            gemini_chat = gemini_model.start_chat(history=[])
             
-            response = chat.send_message('give next small response for this conversation like a doctor.  '+str(chain))
-            return response.text
+            gemini_response = gemini_chat.send_message('give next small response for laste patient response like a doctor.  '+str(chain))
+            return gemini_response.text
         except:
+            print("\n llmmaa ===================\n")
             result_ex = qa_chain(sys+chain+"""\n\n\nalways give small and single response based on the patient 
                 response. don't ask any question give simple response""")
             if "Patient:" in str(result_ex['result']) or "Patient response:" in str(result_ex['result']) or "Patient Response" in str(result_ex['result']):