Error Fix

app.py

@@ -208,14 +208,26 @@ def process_file_upload(files):
                 cursor.execute(f"SELECT COUNT(*) FROM {table_name};")
                 row_count = cursor.fetchone()[0]
                 
+                # Get sample of data
+                cursor.execute(f"SELECT * FROM {table_name} LIMIT 5;")
+                sample_rows = cursor.fetchall()
+                
                 conn.close()
                 
-                file_info.append(f"CSV data loaded into table: {table_name}")
-                file_info.append(f"Columns: {', '.join(columns)}")
-                file_info.append(f"Rows: {row_count}")
+                file_info.append("✅ CSV File Successfully Loaded")
+                file_info.append(f"📊 Table Name: {table_name}")
+                file_info.append(f"📈 Total Rows: {row_count:,}")
+                file_info.append(f"\n📋 Columns:")
+                for col in columns:
+                    file_info.append(f"  • {col}")
+                
+                if sample_rows:
+                    file_info.append("\n🔍 Sample Data (first 5 rows):")
+                    sample_df = pd.DataFrame(sample_rows, columns=[col.split(' ')[0] for col in columns])
+                    file_info.append(f"```\n{sample_df.to_string()}\n```")
                 
             except Exception as e:
-                file_info.append(f"Error loading CSV {file_name}: {str(e)}")
+                file_info.append(f"❌ Error loading CSV {file_name}: {str(e)}")
         
         else:
             # Process PDF or other document types
@@ -229,9 +241,12 @@ def process_file_upload(files):
                     "table_name": None
                 }
                 
-                file_info.append(f"{result['message']} ({result['chunks']} chunks)")
+                file_info.append("✅ Document Successfully Processed")
+                file_info.append(f"📄 File: {file_name}")
+                file_info.append(f"📚 Chunks: {result['chunks']}")
+                file_info.append(result['message'])
             except Exception as e:
-                file_info.append(f"Error processing document {file_name}: {str(e)}")
+                file_info.append(f"❌ Error processing document {file_name}: {str(e)}")
     
     return "\n".join(file_info)
 
@@ -280,15 +295,9 @@ def load_csv_to_sqlite(file_path, conn, table_name):
 
 def list_documents():
     """List all indexed documents"""
-    docs = document_assistant.get_all_documents()
-    if not docs:
-        return "No documents indexed yet"
-    
-    doc_list = []
-    for doc in docs:
-        doc_list.append(f"{doc['filename']} (ID: {doc['id']})")
+    info_list = []
     
-    # Also list CSV tables
+    # Check for CSV data
     try:
         conn = sqlite3.connect(DB_PATH)
         cursor = conn.cursor()
@@ -296,7 +305,7 @@ def list_documents():
         tables = cursor.fetchall()
         
         if tables:
-            doc_list.append("\nCSV data tables:")
+            info_list.append("📊 CSV Data Tables:")
             for table in tables:
                 # Get column info
                 cursor.execute(f"PRAGMA table_info({table[0]});")
@@ -306,13 +315,38 @@ def list_documents():
                 cursor.execute(f"SELECT COUNT(*) FROM {table[0]};")
                 row_count = cursor.fetchone()[0]
                 
-                doc_list.append(f"- {table[0]} ({row_count} rows, {len(columns)} columns)")
+                # Get sample of unique values for some interesting columns
+                sample_info = []
+                for col in ['vendor_id', 'rate_code', 'payment_type']:
+                    if col in columns:
+                        cursor.execute(f"SELECT DISTINCT {col} FROM {table[0]} LIMIT 5;")
+                        unique_vals = [str(row[0]) for row in cursor.fetchall()]
+                        if unique_vals:
+                            sample_info.append(f"{col}: {', '.join(unique_vals)}")
+                
+                info_list.append(f"\n🔹 Table: {table[0]}")
+                info_list.append(f"  - Rows: {row_count:,}")
+                info_list.append(f"  - Columns: {len(columns)}")
+                if sample_info:
+                    info_list.append("  - Sample values:")
+                    for info in sample_info:
+                        info_list.append(f"    • {info}")
         
         conn.close()
     except Exception as e:
-        doc_list.append(f"Error listing CSV tables: {str(e)}")
+        info_list.append(f"Error accessing CSV data: {str(e)}")
+    
+    # Check for indexed documents
+    docs = document_assistant.get_all_documents()
+    if docs:
+        info_list.append("\n📑 Indexed Documents:")
+        for doc in docs:
+            info_list.append(f"- {doc['filename']} (ID: {doc['id']})")
+    
+    if not info_list:
+        return "No data or documents loaded yet"
     
-    return "\n".join(doc_list)
+    return "\n".join(info_list)
 
 def clear_context():
     """Clear the current context and chat history"""