Merge pull request #5 from pradeepgudipati/bug_fixes_22_Apr

Bug fixes 22 apr

.gitignore

@@ -132,3 +132,4 @@ cython_debug/
 *.jsonl
 *.csv
 *.ann
+/chroma_db

README.md

@@ -136,8 +136,8 @@ This python code converts the jsonl data to a sqllite database with a table call
 The messages table has 3 rows
 
 1. id - Primary key
-2. User - Contains the Question
-3. Assistant - Contains the answer to the Question
+2. Question - Contains the Question
+3. Answer - Contains the answer to the Question
 
 ### 2. DB to JSONL 
  - SQLite to JSONL Converter code - [db_to_jsonl.py](jsonl_data_to_db%2Fdb_to_jsonl.py)

src/app.py

@@ -175,7 +175,7 @@ def add_new_qa():
     :return: response
     """
     data = request.json
-    new_item = LLMDataModel(user=data["question"], assistant=data["answer"])
+    new_item = LLMDataModel(question=data["question"], answer=data["answer"])
     db.session.add(new_item)
     db.session.commit()
     return jsonify(status="success", message=f"New item added."), 200

src/frontend/static/css/table_main.css

@@ -5,6 +5,7 @@
 /* CSS class for CKEditor field in edit mode */
 .editing {
     background-color: cornsilk !important;
+    border: black 3px solid;
 }
 
 .not-editing {

src/frontend/static/js/table_main.js

@@ -4,10 +4,11 @@ let upload_file_name = "";
 // Toggle Edit or Save Question buttons
 function toggleQuestionEditor(item_id) {
     const questionDiv = document.getElementById('question-' + item_id);
-    const isEditing = questionDiv.getAttribute('contenteditable') === 'true';
+    const innerDiv = questionDiv.querySelector('div');
+    const isEditing = questionDiv.getAttribute('contenteditable');
     const question_btn_text_field = document.getElementById('q-edit-button-text-' + item_id);
     const question_btn_icon_field = document.getElementById('q-edit-button-icon-' + item_id);
-    if (isEditing) {
+    if (isEditing === 'true') {
         // Save the edited question
         let newQuestion = questionDiv.innerText;
         fetch('/api/update_question', {
@@ -27,19 +28,20 @@ function toggleQuestionEditor(item_id) {
             });
 
         questionDiv.setAttribute('contenteditable', 'false');
+        innerDiv.setAttribute('contenteditable', 'false');
         questionDiv.classList.remove('editing');
         question_btn_text_field.innerText = 'Edit Q';
         question_btn_icon_field.src = staticUrl + "res/questionEditIcon.svg";
         questionDiv.classList.add('not-editing');
     } else {
         // Enable editing
         questionDiv.setAttribute('contenteditable', 'true');
+        innerDiv.setAttribute('contenteditable', 'true');
         questionDiv.classList.add('editing');
         questionDiv.focus();
 
         question_btn_text_field.innerText = 'Save';
         question_btn_icon_field.src = staticUrl + "res/saveIcon.svg";
-        questionDiv.focus();
         questionDiv.classList.add('editing');
     }
 }

src/frontend/templates/header.html

@@ -5,4 +5,5 @@
     <link href="https://cdn.jsdelivr.net/npm/tailwindcss@2.2.19/dist/tailwind.min.css" rel="stylesheet">
     <script src="https://cdn.ckeditor.com/4.16.0/standard/ckeditor.js"></script>
     <title>LLM Training Tools</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='css/table_main.css') }}">
 </head>