Update app.py

app.py

@@ -118,18 +118,28 @@ def analyze_data(data, visualization_type, class_size=10):
     prompt = generate_groq_prompt(data, visualization_type, class_size)
     return prompt
 
-# Function to generate a prompt and preview it with a table format
+# Function to generate a prompt with an 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
-    
-    # Create the textual prompt
+    data_snippet = data.head(10)
+
+    # Convert the data to a string table format with proper indentation
+    table_string = data_snippet.to_markdown(index=False, tablefmt="grid")
+
+    # Create the prompt text with the table included
     prompt = f"""
     The user has uploaded a dataset and selected the '{visualization_type}' visualization type with a class size of {class_size}.
     Below is a preview of the dataset:
+
+    ```
+    {table_string}
+    ```
+
+    Please generate Python code that achieves the requested visualization. For data, please write it directly in the code rather than using file input.
     """
 
-    return prompt, data_snippet
+    return prompt
+
 
 
 # Streamlit App
@@ -156,15 +166,12 @@ if uploaded_file is not None:
             class_size = st.slider("Select the class size for certain plots (e.g., Histogram)", 5, 50, 10)
 
             # Perform Analysis and Visualization
-            prompt, data_snippet = analyze_data(data, visualization_type, class_size)
+            prompt = analyze_data(data, visualization_type, class_size)
             
-            # Display the textual part of the prompt
+            # Display the textual prompt with the indented table
             st.subheader("Prompt Sent to Groq")
-            st.text(prompt)
-            
-            # Display the data preview as a table
-            st.subheader("Preview of the Dataset (as Table)")
-            st.table(data_snippet)
+            st.text_area("Preview of Prompt", value=prompt, height=300)
+
 
             # Chat with Groq Section
             st.subheader("Chat with Groq")