Update app.py

app.py

@@ -229,7 +229,7 @@ def Web_Search(history):
         source_knowledge = ""
 
     augmented_prompt = f"""
-    If you can't answer the next Query using the Search Content, say 'No Answer Is Available' and then just give guidance for the query.
+    You are an AI designed to help understand and extract information from provided Search Content. Based on the user's Query, you may need to summarize the text, answer specific questions, or provide guidance if the answer isn't available.
     Query: {message}
     Search Content:
     {source_knowledge}
@@ -257,27 +257,40 @@ def Chart_Generator(history):
     ]
 
     message = history[-1][0]
-    chart_url = f"https://quickchart.io/natural/{message}"
-    response = requests.get(chart_url)
-
-    if response.status_code == 200:
-        image_html = f'<img src="{chart_url}" alt="Generated Chart" style="display: block; margin: auto; max-width: 100%; max-height: 100%;">'
-        message_with_description = f"Describe and analyse the content of this chart: {chart_url}"
-
-        prompt = HumanMessage(content=message_with_description)
+    if '#chart' in message:
+        message = user_input.split('#chart', 1)[1].strip()
+        chart_url = f"https://quickchart.io/natural/{message}"
+        response = requests.get(chart_url)
+
+        if response.status_code == 200:
+            image_html = f'<img src="{chart_url}" alt="Generated Chart" style="display: block; margin: auto; max-width: 100%; max-height: 100%;">'
+            message_with_description = f"Describe and analyse the content of this chart: {chart_url}"
+    
+            prompt = HumanMessage(content=message_with_description)
+            messages.append(prompt)
+    
+            response = chat_model.invoke(messages)
+            messages.append(AIMessage(content=response.content))
+    
+            if len(messages) >= 8:
+                messages = messages[-8:]
+    
+            combined_content = f'{image_html}<br>{response.content}'
+        else:
+            response_text = "Can't generate this image. Please provide valid chart details."
+            combined_content = response_text
+    else:
+        prompt = HumanMessage(content=message)
         messages.append(prompt)
-
+    
         response = chat_model.invoke(messages)
         messages.append(AIMessage(content=response.content))
-
+    
         if len(messages) >= 8:
             messages = messages[-8:]
 
-        combined_content = f'{image_html}<br>{response.content}'
-    else:
-        response_text = "Can't generate this image. Please provide valid chart details."
-        combined_content = response_text
-
+        combined_content=response.content
+        
     history[-1][1] = ""
     for character in combined_content:
         history[-1][1] += character
@@ -313,7 +326,7 @@ def Link_Scratch(history):
             response_message = "The provided link is empty or does not contain any meaningful words."
         else:
             augmented_prompt = f"""
-            If you can't answer the next Query using the Link Content, say 'No Answer Is Available' and then just give guidance for the query.
+            You are an AI designed to help understand and extract information from provided Link Content. Based on the user's Query, you may need to summarize the text, answer specific questions, or provide guidance if the answer isn't available.
             Query: {user_message}
             Link Content:
             {content_data}
@@ -372,7 +385,7 @@ def File_Interact(history,filepath):
             response_message = "The file is empty or does not contain any meaningful words."
         else:
             augmented_prompt = f"""
-            If you can't answer the next Query using the File Content, say 'No Answer Is Available' and then just give guidance for the query.
+            You are an AI designed to help understand and extract information from provided File Content. Based on the user's Query, you may need to summarize the text, answer specific questions, or provide guidance if the answer isn't available.
             Query: {user_message}
             File Content:
             {content_data}
@@ -444,7 +457,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
     with gr.Row():
-      msg = gr.Textbox(show_label=False, placeholder="Type a message...", scale=10, container=False)
+      msg = gr.Textbox(show_label=False, placeholder="To generate chart: #chart [your message]. To discuss the chart: type your message.", scale=10, container=False)
       submit = gr.Button("➡️Send", scale=1)
 
     clear = gr.ClearButton([msg, chatbot])