Update app.py

app.py

@@ -6,7 +6,9 @@ import seaborn as sns
 import matplotlib.pyplot as plt
 import gradio as gr
 
+# Set your OpenAI API key
 openai.api_key = 'sk-proj-PMkGJxtGRdaihzh15yJYT3BlbkFJ0bEWbrsZjjwV5d3XYSFc'
+
 def load_file(file):
     file_type = file.name.split('.')[-1]
     if file_type == 'csv':
@@ -30,41 +32,8 @@ def load_doc(file):
     doc = docx.Document(file.name)
     text = "\n".join([para.text for para in doc.paragraphs])
     return pd.DataFrame({"text": [text]})
-    def generate_query(prompt):
-    response = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt=prompt,
-        max_tokens=150
-    )
-    return response.choices[0].text.strip()
-
-def handle_query(query, df):
-    if "number of columns" in query.lower():
-        return f"The number of columns is {df.shape[1]}"
-    elif "number of rows" in query.lower():
-        return f"The number of rows is {df.shape[0]}"
-    else:
-        try:
-            # Try executing the query as a pandas query
-            result_df = df.query(query)
-            return result_df.to_html()
-        except Exception as e:
-            return str(e)
 
-def draw_chart(query, df):
-    try:
-        result_df = df.query(query)
-        sns.scatterplot(data=result_df, x=result_df.columns[0], y=result_df.columns[1])
-        plt.title("Generated Chart")
-        plt.xlabel(result_df.columns[0])
-        plt.ylabel(result_df.columns[1])
-        plt.savefig('/content/chart.png')
-        plt.close()
-        return '/content/chart.png'
-    except Exception as e:
-        return str(e)
-
-        def generate_query(prompt):
+def generate_query(prompt):
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
         messages=[
@@ -99,7 +68,7 @@ def draw_chart(query, df):
     except Exception as e:
         return str(e)
 
-        def chatbot(file, input_text):
+def chatbot(file, input_text):
     try:
         # Load the file into a DataFrame
         df = load_file(file)
@@ -120,15 +89,14 @@ def draw_chart(query, df):
     except Exception as e:
         return None, str(e)
 
-        # Create a Gradio interface
+# Create a Gradio interface
 iface = gr.Interface(
     fn=chatbot,
-    inputs=[gr.File(type="filepath", label="Upload File"), gr.Textbox(lines=2, placeholder="Enter your query here...")],
+    inputs=[gr.File(type="file", label="Upload File"), gr.Textbox(lines=2, placeholder="Enter your query here...")],
     outputs=["image", "html"],
     title="Data Analyst Chatbot",
     description="Upload a file and enter a query to get responses based on the data."
 )
 
 # Launch the interface
-# Launch the interface
-iface.launch(share=True, debug=True)
+iface.launch()