Update app.py

app.py

@@ -1,8 +1,6 @@
 import gradio as gr
 import openai
 import pandas as pd
-import matplotlib.pyplot as plt
-import seaborn as sns
 
 # Load your CSV file
 data = pd.read_csv("RR.csv")
@@ -31,52 +29,25 @@ def get_insights(question):
     answer = query_gpt(prompt)
     return answer
 
-def create_visualization_report():
-    # Customize your visualizations, for example:
-    # Plotting a heatmap to show correlations between columns
-    fig, ax = plt.subplots(figsize=(10, 8))
-    sns.heatmap(data.corr(), annot=True, fmt=".2f", cmap="coolwarm", ax=ax)
-    plt.title("Correlation Heatmap")
-
-    # Save the visualization to a file
-    fig.savefig("visualization_report.png")
-    plt.close(fig)
-
-    # Return the saved file
-    return "visualization_report.png"
-
 markdown_data = f"Mock Dataset For GDS Social MBR :\n```\n{data.head(10).to_string(index=False)}\n```\n"
 
-def gradio_wrapper(question, generate_visualization):
-    if generate_visualization:
-        return create_visualization_report()
-    else:
-        return get_insights(question)
-
 iface = gr.Interface(
-    fn=gradio_wrapper,
+    fn=get_insights,
     inputs=[
         gr.inputs.Textbox(lines=2, label="Enter your question"),
-        gr.inputs.Checkbox(label="Generate Visualization Report"),
     ],
-    outputs=[
-        gr.outputs.Textbox(label="Answer"),
-        gr.outputs.Image(type="filepath", label="Visualization Report"),
-    ],
-    title="GPT-powered Q&A with Visualization",
+    outputs="text",
+    title="GPT-powered Q&A",
     description=markdown_data,
     examples=[
-    ["Are there any trends or seasonality in calls handled?", False],
-    ["Are there any patterns that suggest a need for additional staff during specific periods?", False],
-    ["Can you identify any months with a significantly higher or lower number of calls handled compared to the overall average?", False],
-    ["Are there any patterns that suggest a need for additional staff during specific periods?", False],
-    ["", True],
-],
-
+        ("Are there any trends in sales forecast or actual sales?",),
+        ("How does the marketing budget correlate with the sales actual?",),
+        ("Can you identify any months with a significantly higher or lower number of support chats or calls compared to the overall average?",),
+        ("Are there any patterns that suggest a need for additional staff during specific periods based on support chats and calls?",),
+    ],
     allow_screenshot=False,
     theme="compact",
     layout="vertical",
 )
 
 iface.launch(inbrowser=True)
-