Update app.py

app.py

@@ -1,6 +1,8 @@
 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")
@@ -29,22 +31,46 @@ 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=get_insights,
+    fn=gradio_wrapper,
     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="file", label="Visualization Report"),
     ],
-    outputs="text",
-    title="GPT-powered Q&A",
+    title="GPT-powered Q&A with Visualization",
     description=markdown_data,
     examples=[
-        ("Are there any trends or seasonality in calls handled?",),
-        ("Are there any patterns that suggest a need for additional staff during specific periods?",),
-        ("Can you identify any months with a significantly higher or lower number of calls handled compared to the overall average?",),
-    ("Are there any patterns that suggest a need for additional staff during specific periods?",),
+        ("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),
     ],
     allow_screenshot=False,
     theme="compact",