Update database.py

database.py

@@ -74,14 +74,31 @@ def get_table_schema():
     """
     Returns the current table schema as a string.
     """
-    if 'data_table' not in metadata_obj.tables:
-        return None
+    try:
+        # Get list of tables
+        with engine.connect() as con:
+            tables = con.execute(text(
+                "SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%'"
+            )).fetchall()
+            
+        if not tables:
+            return None
+            
+        # Use the first table found
+        table_name = tables[0][0]
         
-    table = metadata_obj.tables['data_table']
-    schema = []
-    for column in table.columns:
-        col_type = str(column.type).upper()
-        is_primary = "primary key" if column.primary_key else ""
-        schema.append(f"- {column.name} ({col_type}) {is_primary}".strip())
-    
-    return "\n".join(schema)
+        # Get column information
+        with engine.connect() as con:
+            columns = con.execute(text(f"PRAGMA table_info({table_name})")).fetchall()
+            
+        schema = []
+        for col in columns:
+            col_name = col[1]
+            col_type = col[2]
+            is_primary = "primary key" if col[5] == 1 else ""
+            schema.append(f"- {col_name} ({col_type}) {is_primary}".strip())
+        
+        return "\n".join(schema)
+        
+    except Exception as e:
+        return None