Update app.py

app.py

@@ -11,16 +11,18 @@ import time
 import shutil
 import json
 import nltk
+#commenting audio related code based on Arun's input
 # audio package
-import speech_recognition as sr
+"""import speech_recognition as sr
 from pydub import AudioSegment
-from pydub.playback import play
+from pydub.playback import play"""
+# Commenting SMTP code since HFSpaces doesn't support it
 # email library
-import smtplib, ssl
+"""import smtplib, ssl
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
 from email.mime.base import MIMEBase
-from email import encoders
+from email import encoders"""
 # langchain
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
@@ -51,15 +53,20 @@ load_dotenv()
 # langfuse analytics
 from langfuse.callback import CallbackHandler
 
+# Inventory API data table
+from tabulate import tabulate
+
 #forcefully stop the agent execution
 import concurrent.futures
+import threading
 
-from tabulate import tabulate
+# mailjet_rest to send email
+from mailjet_rest import Client
+import base64
 
-# Define global variables for managing the thread and future
+# Define global variables for managing the thread and current_event
 executor = concurrent.futures.ThreadPoolExecutor(max_workers=1)
-future = None
-import threading
+current_event = None
 stop_event = threading.Event()
 
 # LangFuse API keys and host settings
@@ -81,9 +88,11 @@ db_uri = os.getenv("POSTGRESQL_CONNECTION")
 # Database setup
 
 db = SQLDatabase.from_uri(db_uri)
+
 user_email = ""
 warehouse_name = ""
 warehouse_id = ""
+# Today's date to be populated in inventory API
 inventory_date = datetime.today().strftime('%Y-%m-%d')
 apis = [
     # fetch warehouse ID
@@ -193,9 +202,6 @@ def get_ASN_data(question):
     except Exception as err:
         print(f"An error occurred: {err}")
 
-
-# get_ASN_data("ASN24072400001")
-
 def load_and_split_pdf(pdf_path):
     reader = PdfReader(pdf_path)
     text = ''
@@ -248,15 +254,12 @@ def document_data_tool(question):
     # summarized_response = summarize_document(query_response)
     return query_response
 
-#mailjet API
-def send_email_with_attachment_mailjet(recipient_email, subject, body, attachment_path):
-    from mailjet_rest import Client
-    import os
-    import base64
-    
-    # Replace these with your Mailjet API credentials
-    api_key = '9ce0aa1f8eb21d9e2ed2433a5702185b'
-    api_secret = '434e366db9b8c421eef99f6f92129fad'
+# mailjet API since SMTP is not supported HF spaces
+def send_email_with_attachment_mailjet(recipient_email, subject, body, attach_img_base64):
+
+
+    api_key = os.getenv("MAILJET_API_KEY")
+    api_secret = os.getenv("MAILJET_API_SECRET")
     
     # Initialize the Mailjet client
     mailjet = Client(auth=(api_key, api_secret), version='v3.1')
@@ -283,7 +286,7 @@ def send_email_with_attachment_mailjet(recipient_email, subject, body, attachmen
                     {
                         "ContentType": "image/png",  # Replace with the correct MIME type of your image
                         "Filename": "inventory_report.png",  # Name of the image as it will appear in the email
-                        "Base64Content": attachment_path  # Base64-encoded image content
+                        "Base64Content": attach_img_base64  # Base64-encoded image content
                     }
                 ]
     
@@ -293,7 +296,7 @@ def send_email_with_attachment_mailjet(recipient_email, subject, body, attachmen
     
     # Send the email
     result = mailjet.send.create(data=data)
-    print(result)
+
     # Check if the email was sent successfully
     if result.status_code == 200:
         print("Email sent successfully with attachment!")
@@ -381,40 +384,43 @@ def inventory_report(question):
             if "warehouseId" in api["params"]:
                 api["params"]["warehouseId"] = warehouse_id
 
-    data1 = make_api_request(apis[1]["url"], apis[1]["params"])
-
-    headers = ["S.No", "Warehouse Code", "Warehouse Name", "Customer Code", "Customer Name", "Item Code", "Item Name",
-               "Currency", "EAN", "UOM", "Quantity", "Gross Weight", "Volume", "Total Value"]
-    table_data = []
 
-    for index, item in enumerate(data1['result'], start=1):
-        row = [
-            index,  # Serial number
-            item['warehouse']['code'],
-            item['warehouse']['name'],
-            item['customer']['code'],
-            item['customer']['name'],
-            item['skuMaster']['code'],
-            item['skuMaster']['name'],
-            item['currency']['code'],
-            item['eanUpc'],
-            item['uom']['code'],
-            item['totalQty'],
-            item['grossWeight'],
-            item['volume'],
-            item['totalValue']
-        ]
-        table_data.append(row)
+    data1 = make_api_request(apis[1]["url"], apis[1]["params"])
+    if (data1):
+        headers = ["S.No", "Warehouse Code", "Warehouse Name", "Customer Code", "Customer Name", "Item Code", "Item Name",
+                   "Currency", "EAN", "UOM", "Quantity", "Gross Weight", "Volume", "Total Value"]
+        table_data = []
+
+        for index, item in enumerate(data1['result'], start=1):
+            row = [
+                index,  # Serial number
+                item['warehouse']['code'],
+                item['warehouse']['name'],
+                item['customer']['code'],
+                item['customer']['name'],
+                item['skuMaster']['code'],
+                item['skuMaster']['name'],
+                item['currency']['code'],
+                item['eanUpc'],
+                item['uom']['code'],
+                item['totalQty'],
+                item['grossWeight'],
+                item['volume'],
+                item['totalValue']
+            ]
+            table_data.append(row)
 
-    # Convert to pandas DataFrame
-    df = pd.DataFrame(table_data, columns=headers)
+        # Convert to pandas DataFrame
+        df = pd.DataFrame(table_data, columns=headers)
 
-    sdf = SmartDataframe(df, config={"llm": llm_chart})
+        sdf = SmartDataframe(df, config={"llm": llm_chart})
 
-    # chart = sdf.chat("Can you draw a bar chart with all avaialble item name and quantity.")
-    chart = sdf.chat(question)
+        # chart = sdf.chat("Can you draw a bar chart with all avaialble item name and quantity.")
+        chart = sdf.chat(question)
 
-    return chart
+        return chart
+    else:
+        return "There are no inventory details for the warehouse you have given."
 
 
 # inventory_report("WH:can you give me a bar chart with item name and quantity for the warehouse WH")
@@ -506,39 +512,39 @@ def clean_gradio_tmp_dir():
 max_iterations = 5
 iterations = 0
 
+
 def handle_query(user_question, chatbot, audio=None):
 
     """
     Function to handle the processing of user input with `AgentExecutor.invoke()`.
     """
-    global future, stop_event
+    global current_event, stop_event
 
-    # Clear previous stop event and future
+    # Clear previous stop event and current_event
     stop_event.clear()
-    print(user_question)
-    #if future and not future.done():
+
+    #if current_event and not current_event.done():
      #   chatbot.append(("","A query is already being processed. Please stop it before starting a new one."))
       #  return gr.update(value=chatbot)
        
-    print(user_question)
     # Start the processing in a new thread
-    future = executor.submit(answer_question_thread, user_question, chatbot)
+    current_event = executor.submit(answer_question_thread, user_question, chatbot)
     
-    # Periodically check if future is done
-    while not future.done():
+    # Periodically check if current_event is done
+    while not current_event.done():
         if stop_event.is_set():
-            future.cancel()
+            current_event.cancel()
             chatbot.append((user_question, "Sorry, we encountered an error while processing your request. Please try after some time."))
             return gr.update(value=chatbot)
             
         time.sleep(1)  # Wait for 1 second before checking again
 
-    if future.cancelled():
+    if current_event.cancelled():
         chatbot.append((user_question, "Sorry, we encountered an error while processing your request. Please try after some time."))
         return gr.update(value=chatbot)
     else:
         try:
-            user_question1, response_text1 = future.result()  # Get the result of the completed future
+            user_question1, response_text1 = current_event.result()  # Get the result of the completed current_event
             print("output")
             print(user_question1) 
             print(response_text1)
@@ -554,10 +560,10 @@ def stop_processing(chatbot):
     """
     Stops the current processing if it's running.
     """
-    global future, stop_event
-    if future and not future.done():
+    global current_event, stop_event
+    if current_event and not current_event.done():
         stop_event.set()  # Signal the process to stop
-        future.cancel()  # Attempt to cancel the future
+        current_event.cancel()  # Attempt to cancel the current_event
     chatbot.append((user_question, "Sorry, we encountered an error while processing your request. Please try after some time."))
     return gr.update(value=chatbot)
 
@@ -571,6 +577,7 @@ def answer_question_thread(user_question, chatbot, audio=None):
     # Clean the /tmp/gradio/ directory
     # clean_gradio_tmp_dir()
     # Handle audio input if provided
+    """
     if audio is not None:
         sample_rate, audio_data = audio
         audio_segment = AudioSegment(
@@ -592,6 +599,7 @@ def answer_question_thread(user_question, chatbot, audio=None):
             user_question = "Sorry, I could not understand the audio."
         except sr.RequestError:
             user_question = "Could not request results from Google Speech Recognition service."
+        """
 
     while iterations < max_iterations:
 
@@ -626,8 +634,7 @@ def answer_question_thread(user_question, chatbot, audio=None):
         img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
 
         img = f'<img src="data:image/png;base64,{img_str}" style="width:450px; height:400px;">'
-        # image = gr.Image(value=img_str)
-        #chatbot.append((user_question, img))
+
         response_text = response.get("output", "").split(".")[0] + img
 
         email_pattern = r'[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}'
@@ -643,17 +650,8 @@ def answer_question_thread(user_question, chatbot, audio=None):
                     subject="Warehouse Inventory Report",
                     body=response.get("output", "").split(".")[0],
                     # attachment_path=chart_path
-                    attachment_path=img_str)
+                    attach_img_base64=img_str)
                 
-                # Send email with the chart image attached
-                """send_email_with_attachment(
-                    recipient_email=user_email,
-                    subject="Warehouse Inventory Report",
-                    body=response.get("output", "").split(":")[0],
-                    # attachment_path=chart_path
-                    attachment_path=os.getenv("IMAGE_PATH")
-                )"""
-        #return user_question, img
 
         if "send email to" in user_question:
             try:
@@ -663,22 +661,15 @@ def answer_question_thread(user_question, chatbot, audio=None):
             except Exception as e:
                 print(f"Error loading image file: {e}")
                 response_text =  "Chart generation failed. Please try again."
-                #chatbot.append((user_question, "Chart generation failed. Please try again."))
-        #return gr.update(value=chatbot)
-        return user_question, response_text
-
 
-    # return [(user_question,gr.Image("/home/user/app/exports/charts/temp_chart.png"))]
-    #  return "/home/user/app/exports/charts/temp_chart.png"
+        return user_question, response_text
     else:
-        #chatbot.append((user_question, response_text))
-        #return gr.update(value=chatbot)
         return user_question, response_text
     # response_text = response_text.replace('\n', ' ').replace('  ', ' ').strip()
     # return response_text
 
 
-
+# without forceful stop option
 def answer_question(user_question, chatbot, audio=None):
      
     global iterations
@@ -803,6 +794,7 @@ def handle_dislike(data: gr.LikeData):
         print("upvote")
         return gr.update(visible=False), gr.update(visible=False)
 
+# greet with user name on successful login
 def update_message(request: gr.Request):
     return f"<h2 style=' font-family: Calibri;'>Welcome, {request.username}</h4>"
 
@@ -902,4 +894,5 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     sample_button3.click(handle_query, [sample_button3, chatbot], [chatbot])
     sample_button4.click(handle_query, [sample_button4, chatbot], [chatbot])
 
+#user_details for login page
 demo.launch(auth=[("lakshmi", "redmind"), ("arun", "redmind")])