begin work on suggesting data types

liberty-rising · Jan 27, 2024 · e58d983 · e58d983
1 parent 7fcaedd
commit e58d983
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 8 deletions.
diff --git a/backend/llms/system_message_manager.py b/backend/llms/system_message_manager.py
@@ -4,6 +4,10 @@ def __init__(self):
             "analytics_chat": """
                 You are an analytics assistant.
                 You will be generating SQL queries, and providing useful information for reports and analytics based on the given prompt.""",
+            "column_type_suggestion": """
+                You are a column type suggestion assistant.
+                You will be suggesting PostgreSQL column types based on the given prompt.
+            """,
             "sql_code": """
                 You are a PostgreSQL SQL statement assistant.
                 Generate PostgreSQL SQL statements based on the given prompt.

diff --git a/frontend/src/pages/upload/CreateDataProfileWindow.jsx b/frontend/src/pages/upload/CreateDataProfileWindow.jsx
@@ -55,7 +55,18 @@ function CreateDataProfileWindow({ open, onClose, onCreate }) {
   };
 
   return (
-    <Dialog open={open} onClose={onClose} maxWidth="md" fullWidth>
+    <Dialog
+      open={open}
+      onClose={() => {
+        onClose();
+        setName(""); // Reset name
+        setExtractInstructions(""); // Reset extractInstructions
+        setSampleFiles([]); // Reset sampleFiles
+        setPreviewData(null); // Reset previewData
+      }}
+      maxWidth="md"
+      fullWidth
+    >
       <DialogTitle>Create a Data Profile</DialogTitle>
       <DialogContent>
         <form onSubmit={handleSubmit}>

diff --git a/frontend/src/pages/upload/DataPreviewAndSchemaEditor.jsx b/frontend/src/pages/upload/DataPreviewAndSchemaEditor.jsx
@@ -26,7 +26,12 @@ function DataPreviewAndSchemaEditor({ previewData }) {
   useEffect(() => {
     if (data && data.length > 0) {
       const newColumnNames = Object.keys(data[0]);
-      const newColumnTypes = newColumnNames.map(() => "string");
+      let newColumnTypes;
+      if (columnTypes.length === 0) {
+        newColumnTypes = newColumnNames.map(() => "text");
+      } else {
+        newColumnTypes = columnTypes;
+      }
       if (JSON.stringify(newColumnNames) !== JSON.stringify(columnNames)) {
         setColumnNames(newColumnNames);
       }
@@ -78,7 +83,7 @@ function DataPreviewAndSchemaEditor({ previewData }) {
               handleColumnTypeChange(index, event.target.value)
             }
           >
-            <MenuItem value={"string"}>String</MenuItem>
+            <MenuItem value={"text"}>Text</MenuItem>
             <MenuItem value={"number"}>Number</MenuItem>
             <MenuItem value={"boolean"}>Boolean</MenuItem>
             <MenuItem value={"date"}>Date</MenuItem>
@@ -90,11 +95,9 @@ function DataPreviewAndSchemaEditor({ previewData }) {
   };
 
   const handleColumnTypeChange = (index, newType) => {
-    setColumnTypes((prevColumnTypes) => {
-      const newColumnTypes = [...prevColumnTypes];
-      newColumnTypes[index] = newType;
-      return newColumnTypes;
-    });
+    let newColumnTypes = [...columnTypes];
+    newColumnTypes[index] = newType;
+    setColumnTypes(newColumnTypes);
   };
 
   const handleEditClick = (index) => {