Update app.py

app.py

@@ -137,7 +137,7 @@ def run_query(query):
 def database_tool(question):
     # print(question)
     sql_query = generate_sql_query(question)
-    print(sql_query)
+    #print(sql_query)
     return run_query(sql_query)
 
 
@@ -145,8 +145,8 @@ def get_ASN_data(question):
     # print(question)
     base_url = os.getenv("ASN_API_URL")
     complete_url = f"{base_url}branchMaster.id=343&transactionUid={question}&userId=164&transactionType=ASN"
-    print("complete url")
-    print(complete_url)
+    #print("complete url")
+    #print(complete_url)
     try:
         response = requests.get(complete_url)
         data = response.json()
@@ -239,8 +239,8 @@ def document_data_tool(question):
     print(f"Document data tool enter: {question}")
     # query_string = question['tags'][0] if 'tags' in question and question['tags'] else ""
     query_response = query_vector_store(vector_store, question, config={"callbacks": [langfuse_handler]})
-    print("query****")
-    print(query_response)
+    #print("query****")
+    #print(query_response)
     # summarized_response = summarize_document(query_response)
     # print("summary***")
     # print(summarized_response)
@@ -249,10 +249,10 @@ def document_data_tool(question):
 
 def send_email_with_attachment(recipient_email, subject, body, image_data):
     try:
-        print(recipient_email)
-        print(subject)
-        print(body)
-        print(image_data)
+        #print(recipient_email)
+        #print(subject)
+        #print(body)
+        #print(image_data)
         sender_email = os.getenv("EMAIL_SENDER")
         sender_password = os.getenv("EMAIL_PASSWORD")
     
@@ -267,7 +267,7 @@ def send_email_with_attachment(recipient_email, subject, body, image_data):
         """
         # Open the file to be sent
         attachment = open(attachment_path, "rb")
-        print("Attached the image")
+        #print("Attached the image")
         # Instance of MIMEBase and named as p
         part = MIMEBase('application', 'octet-stream')
     
@@ -304,7 +304,7 @@ def send_email_with_attachment(recipient_email, subject, body, image_data):
         server.sendmail(sender_email, recipient_email, text)
         server.quit()
             
-    except error:
+    except Exception as error:
         print(error)
    
     # return 1
@@ -327,7 +327,7 @@ def make_api_request(url, params):
 
 
 def inventory_report(question):
-    print(question)
+    #print(question)
 
 
 
@@ -341,15 +341,15 @@ def inventory_report(question):
     else:
         return "warehouse name not found"
 
-    print(apis[0]["url"])
-    print(apis[0]["params"])
+    #print(apis[0]["url"])
+    #print(apis[0]["params"])
     data = make_api_request(apis[0]["url"], apis[0]["params"])
-    print(warehouse_name)
+    #print(warehouse_name)
     if data:
-        print(data)
+        #print(data)
         # Extracting the id for the warehouse with the name "WH"
         warehouse_id = next((item['id'] for item in data['result'] if item['name'] == warehouse_name), None)
-        print(warehouse_id)
+        #print(warehouse_id)
         if warehouse_id is None:
             return "Please provide a warehouse name available in the database."
         # print(f"The id for the warehouse named {name} is: {warehouse_id}")
@@ -518,7 +518,7 @@ def answer_question(user_question, chatbot, audio=None):
             user_question = "Could not request results from Google Speech Recognition service."
 
     while iterations < max_iterations:
-        print(user_question)
+        #print(user_question)
         """if "send email to" in user_question:
             email_match = re.search(r"send email to ([\w\.-]+@[\w\.-]+)", user_question)
             if email_match:
@@ -528,8 +528,8 @@ def answer_question(user_question, chatbot, audio=None):
                 """
 
         response = agent_executor.invoke({"input": user_question}, config={"callbacks": [langfuse_handler]})
-        print("name")
-        print(warehouse_name)
+        #print("name")
+        #print(warehouse_name)
         if isinstance(response, dict):
             response_text = response.get("output", "")
         else:
@@ -537,10 +537,10 @@ def answer_question(user_question, chatbot, audio=None):
         if "invalid" not in response_text.lower():
             break
         iterations += 1
-    print(os.getenv("IMAGE_PATH"))
-    print(response_text)
-    print("validating. ")
-    print(os.getenv("IMAGE_PATH") in response_text)
+    #print(os.getenv("IMAGE_PATH"))
+    #print(response_text)
+    #print("validating. ")
+    #print(os.getenv("IMAGE_PATH") in response_text)
     if iterations == max_iterations:
         return "The agent could not generate a valid response within the iteration limit."
 
@@ -548,12 +548,12 @@ def answer_question(user_question, chatbot, audio=None):
     if os.getenv("IMAGE_PATH") in response_text:
         # Open the image file
         img = Image.open(os.getenv("IMAGE_PATH"))
-        print(img)
+        #print(img)
         # Convert the PIL Image to a base64 encoded string
         buffered = BytesIO()
         img.save(buffered, format="PNG")
         img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
-        print(img_str)
+        #print(img_str)
         img = f'<img src="data:image/png;base64,{img_str}" style="width:500px; height:400px;">'
         # image = gr.Image(value=img_str)
         chatbot.append((user_question, img))
@@ -562,8 +562,8 @@ def answer_question(user_question, chatbot, audio=None):
         match = re.search(email_pattern, user_question)
         if match:
             user_email =  match.group()  # Return the matched email
-            print("user_email")
-            print(user_email)
+            #print("user_email")
+            #print(user_email)
              # email send
             if user_email:
                 # Send email with the chart image attached