Update app.py

app.py

@@ -120,13 +120,12 @@ def analyze_data(data, visualization_type, class_size=10):
 
 # Function to generate a prompt based on the data analysis
 def generate_groq_prompt(data, visualization_type, class_size):
-    # Convert a small sample of the DataFrame to a string without the index
-    sample_data = data.head(10)  # Include only the first 10 rows for clarity
-    data_without_index = sample_data.to_string(index=False)
+    # Convert DataFrame to a CSV-like string for readability
+    data_snippet = data.head(10).to_csv(index=False)  # Limit to the first 10 rows
 
     prompt = f"""
-    Here is the summary statistics for the dataset (showing a sample of 10 rows for clarity):
-    {data_without_index}
+    Here is a preview of the dataset (in CSV format):
+    {data_snippet}
 
     The user has selected the '{visualization_type}' visualization type with a class size of {class_size}.
     Please generate Python code that does this and for any data, please don't use any file input. Write the data in the code.
@@ -157,13 +156,11 @@ 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
             prompt = analyze_data(data, visualization_type, class_size)
             
-            # Display prompt sent to Groq with preserved formatting
+            # Display the prompt that will be sent to Groq
             st.subheader("Prompt Sent to Groq")
-            st.text_area("Preview of Prompt", prompt, height=200)
-
+            st.code(prompt, language="plaintext")
             # Chat with Groq Section
             st.subheader("Chat with Groq")
             chat_input = st.text_area("Ask Groq questions about the data:")