Upload prompt.py

prompt.py

@@ -0,0 +1,96 @@
+import gradio as gr
+from langchain.agents import create_pandas_dataframe_agent
+from langchain.llms import OpenAI
+import pandas as pd
+from langchain import OpenAI
+from langchain.chat_models import ChatOpenAI
+import os
+from langchain.agents import ZeroShotAgent, Tool, AgentExecutor
+from langchain import OpenAI,LLMChain
+os.environ['OPENAI_API_KEY'] = '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'
+os.environ["WOLFRAM_ALPHA_APPID"] = "5829GX-JVY5Y6LR3Q"
+
+prefix = """
+Hi! I'm a data analysis bot. I can help you perform basic exploratory data analysis (EDA) on your pandas dataframe. 
+
+To get started, please make sure you have pandas and matplotlib installed and provide me with a valid pandas dataframe by assigning it to the variable 'df'. 
+
+if the user ask anything other than questions related to the data analysis answer with "I cannot help you with that"
+When you're ready to begin, type 'Start' and I'll perform the following tasks:
+- List all the column names.
+- Count the number of rows and columns.
+- Count the number of missing values.
+- Compute descriptive statistics for numerical columns (count, mean, standard deviation, minimum, maximum, and quartiles).
+- Get value counts for categorical variables.
+- Identify any outliers, anomalies, or unusual patterns in the data.
+
+
+"""
+
+suffix = """
+        Please make sure to perform all the tasks you can and answer them one by one.\
+        Make sure you perform all the taks you can and answer them one by one
+        answer in this format
+        "list column names:
+        "Index","Class"
+
+Input:
+{input}
+
+Agent Scratchpad:
+{agent_scratchpad}
+"""
+global df
+df = pd.read_csv('titanic.csv')
+pandas = create_pandas_dataframe_agent(OpenAI(temperature=0), df, verbose=True)
+llm = ChatOpenAI(temperature=0)
+tools = [
+    Tool(
+    name = "Pandas",
+    func= pandas.run,
+    description="Useful to seach dataframe and perform data analysis"
+    )
+]
+
+prompt = ZeroShotAgent.create_prompt(
+    tools, 
+    prefix=prefix, 
+    suffix=suffix,
+    input_variables=["input","agent_scratchpad"]
+)
+llm_chain = LLMChain(llm=OpenAI(temperature=0), prompt=prompt)
+
+tool_names = [tool.name for tool in tools]
+agent = ZeroShotAgent(llm_chain=llm_chain, allowed_tools=tool_names)
+
+def reply(input):
+    try:
+        agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True)
+        message = agent_executor.run(input)
+        
+        return message
+    except Exception as e:
+        message = f"An error occurred: {str(e)}"
+    return message
+input_text = gr.inputs.Textbox(
+    label="Enter your questions here",
+    placeholder="Enter 'Start' for basic EDA",
+    lines=3
+    
+)
+text_output = gr.outputs.Textbox(label="Answer")
+
+
+description = ""
+
+iface = gr.Interface(
+                  fn=reply,
+                  inputs=input_text,
+                  outputs=text_output,
+                  title="Auto-Data Analysis",
+                  description=description,
+                  theme="light",
+                  layout="vertical",
+                  allow_flagging=False,
+                 
+                ).launch()