Update app.py

app.py

@@ -46,22 +46,28 @@ def handle_message(message: str, mode: str):
         return "Select a valid mode."
 
 def chat_message(message: str):
+    
     global messages
     prompt = HumanMessage(content=message)
     messages.append(prompt)
+    
     response = chat_model.invoke(messages)
     messages.append(response.content)
+    
     if len(messages) >= 6:
         messages = messages[-6:]
+        
     return f"IT-Assistant: {response.content}"
 
 def web_search(message: str):
+    
     global messages
     similar_docs = db.similarity_search(message, k=3)
     if similar_docs:
         source_knowledge = "\n".join([x.page_content for x in similar_docs])
     else:
         source_knowledge = ""
+        
     augmented_prompt = f"""
     If the answer to the next query is not contained in the Web Search, say 'No Answer Is Available' and then just give guidance for the query.
 
@@ -70,40 +76,37 @@ def web_search(message: str):
     Web Search:
     {source_knowledge}
     """
+    
     prompt = HumanMessage(content=augmented_prompt)
     messages.append(prompt)
+    
     response = chat_model.invoke(messages)
     messages.append(response.content)
+    
     if len(messages) >= 6:
         messages = messages[-6:]
+        
     return f"IT-Assistant: {response.content}"
 
 def chart_generator(message: str):
-    # Construct the full chart URL
-    chart_url = f"https://quickchart.io/natural/{message}"
 
-    # Send a GET request to the chart URL
+    chart_url = f"https://quickchart.io/natural/{message}"
     response = requests.get(chart_url)
 
-    # Check if the request was successful (status code 200)
     if response.status_code == 200:
-        # Add a description to the message
         message_with_description = f"Describe and analyse the content of this chart: {message}"
 
-        # Create a HumanMessage object with the message including the description
         prompt = HumanMessage(content=message_with_description)
         messages.append(prompt)
 
-        # Invoke the chat model with the updated messages
         response = chat_model.invoke(messages)
         messages.append(response.content)
+        
         if len(messages) >= 6:
             messages = messages[-6:]
 
-        # Display the chart image with a smaller size
         display(Image(url=chart_url, width=500, height=300))
 
-        # Return the message and response
         return f"IT-Assistant: {response.content}"
     else:
         return f"Can't generate this image. Please provide valid chart details."