Update app.py

app.py

@@ -3,7 +3,6 @@ from langchain.llms import OpenAI
 from langchain.chat_models import ChatOpenAI
 from langchain.agents.agent_types import AgentType
 from langchain_experimental.agents.agent_toolkits import create_csv_agent
-from langchain_community.chat_models import ChatOpenAI
 import pandas as pd
 import matplotlib.pyplot as plt
 from io import BytesIO
@@ -22,14 +21,15 @@ def create_csv_agent_instance(llm, file_path):
     return agent
 
 # Define the function to perform QA and optionally plot graphs
-def qa_app(csv_file, question, plot=False):
+def qa_app(csv_file, question):
     try:
         df = pd.read_csv(csv_file.name)
-        llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-16k", openai_api_key='sk-proj-bFQSTYKhv8H4NqKfMFuaT3BlbkFJecIT52917IZGzTu8bhia')
+        llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-16k", openai_api_key=key)  # ensure key is properly managed
         agent = create_csv_agent_instance(llm, csv_file.name)
         response = agent.run(question)
 
-        if plot:
+        # Check if the user's question implies a request for a plot
+        if "plot" in question.lower() or "graph" in question.lower():
             graph_output = plot_data(df)
         else:
             graph_output = None
@@ -59,16 +59,16 @@ demo = gr.Interface(
     fn=qa_app,
     inputs=[
         gr.File(label="Upload CSV file"),
-        gr.Textbox(label="Question"),
-        gr.Checkbox(label="Plot Graphs", value=False)
+        gr.Textbox(label="Question")
     ],
     outputs=[
         gr.Textbox(label="Answer"),
-        gr.Image(label="Generated Plot", type="pil"),
+        gr.Image(label="Generated Plot", type="pil", optional=True),  # Mark the plot as optional
     ],
     title="Data Analysis Chatbot",
-    description="Upload a CSV file, ask a question about the data, and optionally plot graphs."
+    description="Upload a CSV file, ask a question about the data. Include 'plot' or 'graph' in your question to generate graphs."
 )
 
 # Launch the Gradio app with debugging enabled to trace any runtime issues
 demo.launch(debug=True)
+