Update app.py

app.py

@@ -3,11 +3,9 @@ import pandas as pd
 import numpy as np
 import seaborn as sns
 import matplotlib.pyplot as plt
-import tempfile
-import subprocess
 from groq import Groq
 
-# Groq API Key setup .
+# Groq API Key setup
 GROQ_API_KEY = "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"
 client = Groq(api_key=GROQ_API_KEY)
 
@@ -19,7 +17,6 @@ def chat_with_groq(prompt):
             model="llama3-8b-8192",
             stream=False
         )
-        print(prompt)
         return chat_completion.choices[0].message.content
     except Exception as e:
         return f"Error fetching response: {e}"
@@ -50,7 +47,7 @@ def parse_file(uploaded_file):
 # Preprocess DataFrame to Fix Type Issues
 def preprocess_dataframe(df):
     try:
-        # Convert problematic columns to string to avoid Arrow serialization issues
+        # Convert problematic columns to string to avoid serialization issues
         for col in df.columns:
             if df[col].dtype.name == 'object' or df[col].dtype.name == 'category':
                 df[col] = df[col].astype(str)
@@ -85,9 +82,6 @@ def analyze_data(data, visualization_type, class_size=10):
         fig, ax = plt.subplots(figsize=(8, 6))
         sns.heatmap(numeric_data.corr(), annot=True, ax=ax, cmap="coolwarm", fmt=".2f")
         st.pyplot(fig)
-
-
-
     elif visualization_type == "Line Graph" and not numeric_data.empty:
         st.subheader("Line Graph")
         x_col = st.selectbox("Select the X-axis column for the Line Graph (Non-Numeric):", numeric_data.columns)
@@ -98,10 +92,6 @@ def analyze_data(data, visualization_type, class_size=10):
         ax.set_xlabel(x_col)
         ax.set_ylabel(y_col)
         st.pyplot(fig)
-
-
-
-
     elif visualization_type == "Area Chart" and not numeric_data.empty:
         st.subheader("Area Chart")
         column = st.selectbox("Select a column for the Area Chart:", numeric_data.columns)
@@ -110,7 +100,6 @@ def analyze_data(data, visualization_type, class_size=10):
         ax.set_xlabel(column)
         ax.set_ylabel("Area")
         st.pyplot(fig)
-
     else:
         st.warning("No valid visualization option selected or data available.")
 
@@ -118,6 +107,7 @@ def analyze_data(data, visualization_type, class_size=10):
     prompt = generate_groq_prompt(data, visualization_type, class_size)
     return prompt
 
+# Generate Groq Prompt with Indented Table
 def generate_groq_prompt(data, visualization_type, class_size):
     # Limit the preview to the first 10 rows for readability
     data_snippet = data.head(10)  # Select top 10 rows for the preview
@@ -150,8 +140,6 @@ def generate_groq_prompt(data, visualization_type, class_size):
     return prompt
 
 
-
-
 # Streamlit App
 st.title("Data Analysis AI")
 st.markdown("Upload a file (CSV or Excel) to analyze it.")
@@ -175,22 +163,14 @@ if uploaded_file is not None:
             # User input for class size customization
             class_size = st.slider("Select the class size for certain plots (e.g., Histogram)", 5, 50, 10)
 
-            # Perform Analysis and Visualization
+            # Perform Analysis and Generate Prompt for Groq
             prompt = analyze_data(data, visualization_type, class_size)
-            
-            # Display the textual prompt with the indented table
-            st.subheader("Prompt Sent to Groq")
-            st.text_area("Preview of Prompt", value=prompt, height=300)
-
 
-            # Chat with Groq Section
-            st.subheader("Chat with Groq")
-            chat_input = st.text_area("Ask Groq questions about the data:")
-            if st.button("Chat"):
-                if chat_input:
-                    chat_response = chat_with_groq(f"Here is the data:\n{data}\n\n{chat_input}")
-                    st.write("Groq's Response:")
-                    st.write(chat_response)
+            # Send the prompt directly to Groq
+            st.subheader("Groq Processing")
+            response = chat_with_groq(prompt)
+            st.write("Groq's Response:")
+            st.write(response)
 
             # Groq Code Generation Section
             st.subheader("Generate Python Code with Groq")
@@ -205,3 +185,4 @@ if uploaded_file is not None:
                 st.code(response, language="python")
     except Exception as e:
         st.error(f"An error occurred: {e}")
+