chore: Update Plotly dependency and improve graph generation in app.py

app.py

@@ -51,7 +51,7 @@ class Agent:
     def execute(self):
         try:
             response = self.client.chat.completions.create(
-                model="meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
+                model="meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
                 stop=["PAUSE"],
                 messages=self.messages,
             )
@@ -86,15 +86,17 @@ class Agent:
 
 
 known_actions = {"runcode": runcode}
-
 prompt = f"""
 You are an expert in writing python code based on user request and you run in a loop of Thought, Action, PAUSE, Observation.
 At the end of the loop you output an Answer
 Use Thought to describe your thoughts about the question you have been asked.
 Use Action to run one of the actions available to you - then return PAUSE.
 Observation will be the result of running those actions.
+
+
+for Graphing library use Plotly and save the plot as graph.html
+Always save generated plot as 'graph.html'
 Always return just code no need of ```
-Always save generated plot as 'graph.png'
 
 Your Task is help user get result of query about below dataset,Decide based on user query to make Plot or Just Textual Answer.
 Here is the metadata of the dataset and name of dataset is customers-10000.csv:
@@ -110,8 +112,7 @@ runcode
 How to use actions
 Action : action_name: input_to_action
 
-if want to use matplotlib use backend as Agg
-
+if input_to_action is code then don't use ``` just write code
 Example session:
 
 Question: Get the average age of the customers
@@ -130,6 +131,7 @@ Observation :  understand the output based its stdout and take necessary steps.
 Answer: Final Answer for User Request if its graph send "Graph Generated" or Textual Answer "Your Interpretation of Answer"
 """.strip()
 
+
 bot = Agent(system_prompt=prompt, known_actions=known_actions)
 
 st.set_page_config(layout="wide")
@@ -154,8 +156,8 @@ with col1:
         if not query:
             st.error("Please enter a query.")
         else:
-            if "graph.png" in os.listdir():
-                os.remove("graph.png")
+            if "graph.html" in os.listdir():
+                os.remove("graph.html")
             st.session_state.logs = ""
             with st.spinner("Generating response..."):
                 bot.query(query)
@@ -164,13 +166,13 @@ with col1:
                 answer_end_index = answer_match.end()
                 subsequent_text = st.session_state.logs[answer_end_index:].strip()
             st.success(subsequent_text)
+
 with col2:
     st.header("Output")
-    if "graph.png" in os.listdir():
-        image = Image.open("graph.png")
-        st.image(image, caption="Generated Plot")
+    if "graph.html" in os.listdir():
+        st.components.v1.html(open("graph.html", "r").read(), height=600)
     else:
         st.write("No plot generated yet.")
 
-st.header("Live Logs")
-st.text_area("Logs", value=st.session_state.logs, height=400, disabled=True)
+with st.expander("Logs"):
+    st.code(st.session_state.logs)

requirements.txt

@@ -1,3 +1,4 @@
 streamlit
 openai
-pillow
+pillow
+plotly