Upload 8 files

.gitattributes

@@ -1,35 +1,35 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,12 +1,12 @@
----
-title: Osuosutesttest
-emoji: 💻
-colorFrom: gray
-colorTo: pink
-sdk: streamlit
-sdk_version: 1.35.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: AI Builder
+emoji: 📈
+colorFrom: blue
+colorTo: gray
+sdk: streamlit
+sdk_version: 1.35.0
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,95 @@
+#python -m streamlit run app.py 
+
+import streamlit as st
+from transformers import pipeline
+from FlagEmbedding import BGEM3FlagModel
+from FlagEmbedding import FlagReranker
+from inference_script import answer_question #import function from another file
+from corpus import corpusvalue
+import numpy as np
+import getpass
+import os
+from langchain.prompts.prompt import PromptTemplate
+from langchain.chains import ConversationChain
+from langchain.chains import LLMChain
+from langchain_google_genai import ChatGoogleGenerativeAI
+import pickle
+import sqlite3
+
+@st.cache_resource
+def load_model():
+    return BGEM3FlagModel('BAAI/bge-m3',use_fp16=True)
+
+@st.cache_resource
+def load_rerank_model():
+    return FlagReranker('BAAI/bge-reranker-v2-m3', use_fp16=True)
+
+@st.cache_resource
+def initLLM():
+    os.environ["GOOGLE_API_KEY"] = "AIzaSyAuKPswmbdM8jCpSt0luez7tjLND-uyY7M"
+    llm = ChatGoogleGenerativeAI(model="gemini-pro")
+    template = """
+    คุณเป็นผู้เชี่ยวชาญด้านกฎหมายจราจร มีหน้าที่ในการนำข้อความทางกฎหมายเเละข้อปฎิบัติเกี่ยวกับการละเมิดกฎจราจรเเละข้อปฎิบัติต่างๆมาตอบคำถามว่าคำถามที่ถามมานั้นว่าผิดหรือไม่หรือจะต้องปฎิบัติตัวอย่างไร เเละอธิบายเพิ่มเติม ให้รายละเอียดและคำอธิบายเพิ่มเติมเพื่อให้ผู้ที่ไม่ใช่ผู้เชี่ยวชาญด้านกฎหมายเข้าใจได้ง่ายขึ้น
+
+    นี้คือคำถาม : {question}
+    ข้อความทางกฎหมาย: {section}
+
+    คำอธิบายโดยละเอียด:
+    """
+    prompt = PromptTemplate(
+        input_variables=["section","question"],
+        template=template
+    )
+    llm_chain = LLMChain(prompt=prompt, llm=llm)
+    return llm_chain
+
+@st.cache_data
+def embeded_corpus():
+    file_path_embeded_corpus = "save/BGM3savesimilar_Corpus" #
+    with open(file_path_embeded_corpus,'rb') as file :
+        BGM3similar_Corpus = pickle.load(file)
+    return BGM3similar_Corpus
+
+def insert_feedback(question, answer,like,dislike, feedback_text):
+    conn = sqlite3.connect('feedback.db')
+    cursor = conn.cursor()
+    cursor.execute('''CREATE TABLE IF NOT EXISTS qa_feedback
+                 (id INTEGER PRIMARY KEY, question TEXT, answer TEXT, 
+                 like INTEGER, dislike INTEGER, feedback_text TEXT)''')
+    data_to_insert = (question, answer, like, dislike, feedback_text)
+    sql_query = 'INSERT INTO qa_feedback (question, answer, like, dislike, feedback_text) VALUES (?, ?, ?, ?, ?)'
+    cursor.execute(sql_query, data_to_insert)
+    conn.commit()
+    conn.close()
+
+model = load_model()
+rerank_model = load_rerank_model()
+llm_chain = initLLM()
+BGM3similar_Corpus = embeded_corpus()
+corpus_list = corpusvalue()
+
+st.title("Traffic Law Question-Answering")
+
+question = st.text_area("Enter your question:")
+
+if 'like_value' not in st.session_state:
+    st.session_state.like_value = 0
+if 'dislike_value' not in st.session_state:
+    st.session_state.dislike_value = 0
+
+if st.button("Get Answer"):
+    if question:
+        answer = answer_question(question=question,model=model,rerankmodel=rerank_model,corpus_embed= BGM3similar_Corpus, corpus_list=corpus_list,llm_chain=llm_chain)
+        st.text_area("Answer:", value=answer, height=500)
+
+        st.write("### Feedback")
+        feedback = st.text_area("Your feedback:")
+        like = st.button("👍 Like")
+        dislike = st.button("👎 Dislike")
+
+        like_value = 1 if like else 0
+        dislike_value = -1 if dislike else 0
+        feedback = feedback if feedback else "No Feed back"
+
+        if like or dislike or feedback:
+            insert_feedback(question, answer,  like_value, dislike_value,feedback)

inference_script.py

@@ -0,0 +1,61 @@
+import numpy as np 
+
+def answer_question(question , model , rerankmodel , corpus_embed , corpus_list,llm_chain):
+    # embeddings_1 = model.encode(question, batch_size=16, max_length=8192 ,)['dense_vecs']
+    # embeddings_2 = corpus_embed
+    # BGM3similarity = embeddings_1 @ embeddings_2.T
+
+#==========================================================
+
+    ALL_final_ans_list_ALL = []
+    batch_size = 10 
+
+    sentence_pairs = [[question, j] for j in corpus_list]
+
+    listofscore = []
+    compute_Score = range(0, len(sentence_pairs), batch_size)
+
+    for i in compute_Score:
+        batch_pairs = sentence_pairs[i:i+batch_size]
+        allscore = model.compute_score(batch_pairs,
+                                        max_passage_length=512,
+                                        weights_for_different_modes=[0.4, 0.2, 0.4]) # sum: w[0]*dense_score + w[1]*sparse_score + w[2]*colbert_score
+        listofscore.append(allscore) 
+
+    score_ALL = []
+
+
+    for score_dict in listofscore:
+
+        score_ALL.extend(score_dict['colbert+sparse+dense'])
+
+    ALL_final_ans_list_ALL.append(score_ALL)
+
+#==========================================================
+
+    topkindex = 15
+    topk15scoresimilar_BGM3 = np.argsort(ALL_final_ans_list_ALL)[:,-topkindex:]
+    # topk15scoresimilar_BGM3 = np.argsort(BGM3similarity)[-topkindex:]
+
+
+    BGM3_1_retrieval = [corpus_list[i] for i in topk15scoresimilar_BGM3[0]]
+
+    scores = []
+
+    for passage in BGM3_1_retrieval:
+        passage = str(passage)
+        score = rerankmodel.compute_score([question, passage], normalize=True)
+        scores.append(score)
+        # print(passage[:20])
+
+    highest_scoring_index = scores.index(max(scores))
+    result_passage = BGM3_1_retrieval[highest_scoring_index]
+    # print(f"Retrieval{result_passage[:20]}")
+    # print(f"Question{question}")
+
+    inputs = {"section": result_passage, "question": question}
+
+    response = llm_chain.run(inputs)
+    print(response)
+    return response
+

requirements.txt

@@ -0,0 +1,7 @@
+numpy==1.26.4
+transformers==4.38.2
+streamlit==1.30.0
+langchain
+db-sqlite3
+langchain_google_genai
+FlagEmbedding 

viewdatabase.ipynb

@@ -0,0 +1,70 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 25,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "(1, 'ขับรถไม่มีป้ายทะเบียน', '**คำถาม:** ขับรถไม่มีป้ายทะเบียน\\n\\n**ข้อความทางกฎหมาย:**\\nมาตรา 7 - ห้ามมิให้ผู้ใดนำรถที่มิได้ติดแผ่นป้ายเลขทะเบียน แผ่นป้ายเครื่องหมายเลขทะเบียนหรือป้ายประจำรถ ตามกฎหมายว่าด้วยรถยนต์ กฎหมายว่าด้วยการขนส่ง กฎหมายว่าด้วยล้อเลื่อน กฎหมายว่าด้วยรถลาก หรือกฎหมายว่าด้วยรถจ้าง มาใช้ในทางเดินรถ\\n\\n**คำอธิบายโดยละเอียด:**\\nขับรถโดยไม่มีป้ายทะเบียนเป็นการกระทำที่ผิดกฎหมาย เนื่องจากป้ายทะเบียนเป็นเครื่องหมายที่ใช้ระบุตัวตนของรถและเจ้าของรถ การนำรถที่ไม่มีป้ายทะเบียนมาใช้ในทางเดินรถถือเป็นความผิดตามกฎหมายว่าด้วยรถยนต์ มีโทษปรับไม่เกิน 2,000 บาท\\n\\n**ข้อปฏิบัติ:**\\nหากถูกเจ้าหน้าที่ตำรวจจับกุมในข้อหาขับรถไม่มีป้ายทะเบียน ผู้ขับขี่จะต้องชำระค่าปรับตามที่กำหนด และหากรถไม่มีป้ายทะเบียนถูกต้องตามกฎหมาย ผู้ขับขี่จะต้องนำรถไปติดแผ่นป้ายทะเบียนที่ถูกต้องก่อนนำมาใช้ในทางเดินรถ', 0, 0, 'No Feed back')\n",
+      "(2, 'ขับรถผ่าไฟสีส้ม', '**คำถาม:** ขับรถผ่าไฟสีส้ม\\n\\n**คำตอบ:** ผิด\\n\\n**ข้อความทางกฎหมาย:**\\nมาตรา 22 (1) สัญญาณจราจรไฟสีเหลืองอำพัน ให้ผู้ขับขี่เตรียมหยุดรถหลังเส้นให้รถหยุดเพื่อเตรียมปฏิบัติตามสัญญาณที่จะปรากฏต่อไปดังกล่าวใน (2)\\n\\n**คำอธิบายเพิ่มเติม:**\\nไฟจราจรสีเหลืองอำพันเป็นสัญญาณเตือนให้ผู้ขับขี่เตรียมหยุดรถ เมื่อเห็นไฟสีเหลืองอำพัน ผู้ขับขี่ต้องชะลอความเร็วและจอดรถหลังเส้นหยุดรถ หากผู้ขับขี่ได้เลยเส้นหยุดรถไปแล้ว จึงสามารถขับรถผ่านไปได้\\n\\nการขับรถผ่าไฟสีเหลืองอำพันถือว่าเป็นการฝ่าฝืนกฎหมายจราจร เนื่องจากผู้ขับขี่ไม่ได้ปฏิบัติตามสัญญาณจราจรอย่างถูกต้อง อาจทำให้เกิดอุบัติเหตุได้', 0, 0, 'No Feed back')\n",
+      "(3, 'ขับรถต้องชิดเลยฝั่งไหน?', '**ผิดหรือไม่:** ไม่ผิด\\n\\n**ข้อปฏิบัติ:**\\n\\nตามกฎหมายจราจร มาตรา 125 กำหนดให้ผู้ขับขี่รถยนต์ต้องขับรถชิดขอบทางด้านซ้ายในกรณีต่อไปนี้:\\n\\n* ขับรถผ่านทางแคบระหว่างภูเขาหรือระหว่างเนิน\\n* ขับรถในทางเดินรถบนภูเขาหรือบนเนิน\\n\\n**คำอธิบายเพิ่มเติม:**\\n\\n* **ทางแคบระหว่างภูเขาหรือระหว่างเนิน:** หมายถึงถนนที่มีความกว��างไม่เพียงพอให้รถยนต์สวนกันได้อย่างปลอดภัย\\n* **ทางเดินรถบนภูเขาหรือบนเนิน:** หมายถึงถนนที่ลาดชันและคดเคี้ยวบนภูเขาหรือเนินเขา\\n\\nเมื่อขับรถในสถานการณ์เหล่านี้ ผู้ขับขี่จะมองเห็นได้ชัดเจนขึ้นและสามารถตอบสนองต่อรถที่สวนมาได้อย่างรวดเร็ว การขับรถชิดซ้ายยังช่วยลดความเสี่ยงที่จะเกิดการชนกับสิ่งกีดขวางข้างทางอีกด้วย\\n\\nนอกจากนี้ เมื่อถึงทางโค้ง ผู้ขับขี่ต้องใช้เสียงสัญญาณเพื่อเตือนรถอื่นที่อาจสวนมา เพื่อความปลอดภัยในขณะขับรถบนภูเขาหรือเนินเขา', 0, 0, 'No Feed back')\n",
+      "(4, 'ขับรถในเวลากลางคืนต้องเปิดไฟหน้าไหม ?', '**คำถาม:** ขับรถในเวลากลางคืนต้องเปิดไฟหน้าไหม?\\n\\n**คำตอบ:** ใช่ ต้องเปิดไฟหน้า\\n\\n**ข้อความทางกฎหมายที่เกี่ยวข้อง:**\\nไม่มีข้อความทางกฎหมายโดยตรงที่ระบุว่าต้องเปิดไฟหน้าในเวลากลางคืน แต่มีข้อความที่เกี่ยวข้องใน\\n\\n* **มาตรา 15 แห่งพระราชบัญญัติจราจรทางบก พ.ศ. 2522** ที่ระบุว่า ในเวลาที่ต้องเปิดไฟตามมาตรา 11 หรือมาตรา 61 ผู้ขับขี่ต้องจุดไฟสัญญาณแสงแดง หรือในเวลากลางวันต้องติดธงสีแดงไว้ที่ตอนปลายสุดของสิ่งที่บรรทุกนั้น โดยจุดไฟสัญญาณหรือติดธงไว้ให้มองเห็นได้ในระยะไม่น้อยกว่าหนึ่งร้อยห้าสิบเมตร\\n\\n* **มาตรา 11 แห่งพระราชบัญญัติจราจรทางบก พ.ศ. 2522** ที่ระบุว่า รถทุกคันที่มีความกว้างเกินกว่า 2.50 เมตร หรือยาวเกินกว่า 6.00 เมตร ต้องติดไฟหน้า ไฟท้าย และไฟเบรกในเวลากลางคืน\\n\\n**ข้อปฏิบัติ:**\\nจากข้อความทางกฎหมายข้างต้น แม้จะไม่มีข้อความที่ระบุโดยตรงว่าต้องเปิดไฟหน้าในเวลากลางคืน แต่เนื่องจากไฟหน้าเป็นอุปกรณ์ที่จำเป็นสำหรับการมองเห็นในเวลากลางคืน เพื่อความปลอดภัยของผู้ขับขี่และผู้ใช้ถนนอื่นๆ จึงควรเปิดไฟหน้าในเวลากลางคืนเสมอ\\n\\n**การละเมิดและโทษ:**\\nการขับรถในเวลากลางคืนโดยไม่เปิดไฟหน้า ถือเป็นการฝ่าฝืนกฎจราจรและอาจถูกปรับตามกฎหมาย', 0, 0, 'No Feed back')\n",
+      "(5, 'ขับรถตอนกลางคืนต้องเปิดไฟหน้าไหม ?', '**คำถาม:** ขับรถตอนกลางคืนต้องเปิดไฟหน้าไหม?\\n\\n**คำตอบ:** ใช่\\n\\n**ข้อความทางกฎหมาย:** มาตรา 61 พระราชบัญญัติจราจรทางบก พ.ศ. 2522\\n\\n**ข้อปฏิบัติ:**\\n\\n* ผู้ขับขี่รถยนต์ทุกชนิดต้องเปิดไฟหน้ารถในระหว่างเวลาตั้งแต่พระอาทิตย์ตกถึงพระอาทิตย์ขึ้น\\n\\n**คำอธิบายเ���ิ่มเติม:**\\n\\nกฎหมายจราจรกำหนดให้ผู้ขับขี่รถยนต์ต้องเปิดไฟหน้ารถในเวลากลางคืนเพื่อเพิ่มทัศนวิสัยให้ทั้งผู้ขับขี่และผู้ใช้ถนนคนอื่นๆ ไฟหน้าช่วยให้ผู้ขับขี่มองเห็นสิ่งกีดขวางและอันตรายอื่นๆ บนท้องถนนได้ชัดเจนขึ้น นอกจากนี้ ยังช่วยให้ผู้ใช้ถนนคนอื่นๆ มองเห็นรถที่กำลังขับขี่อยู่ได้ง่ายขึ้น ซึ่งจะช่วยลดความเสี่ยงต่อการเกิดอุบัติเหตุ\\n\\nการละเมิดกฎหมายนี้ถือเป็นความผิดทางอาญาและอาจส่งผลให้ถูกปรับหรือจำคุกได้', 0, 0, 'No Feed back')\n",
+      "(6, 'ขับรถไม่มีป้ายทะเบียนผิดไหม?', '**ผิด**\\n\\n**ข้อความทางกฎหมาย:** มาตรา 7 แห่ง พระราชบัญญัติรถยนต์ พ.ศ. 2522\\n\\n**คำอธิบายโดยละเอียด:**\\n\\nตามกฎหมายไทย ห้ามมิให้ผู้ใดนำรถที่ไม่ได้ติดแผ่นป้ายทะเบียนมาใช้ในทางเดินรถ การขับขี่รถยนต์ที่ไม่มีป้ายทะเบียนถือเป็นการฝ่าฝืนกฎหมายจราจร ซึ่งอาจนำไปสู่การถูกปรับและดำเนินคดี\\n\\n**เหตุผลที่ต้องติดป้ายทะเบียน:**\\n\\n* เพื่อระบุตัวตนของรถยนต์และเจ้าของรถ\\n* เพื่ออำนวยความสะดวกในการบังคับใช้กฎหมายจราจร\\n* เพื่อช่วยในการติดตามรถยนต์ที่สูญหายหรือถูกขโมย', 0, 0, 'No Feed back')\n",
+      "(7, 'ขับรถไม่มีป้ายทะเบียนผิดไหม?', '**คำถาม:** ขับรถไม่มีป้ายทะเบียนผิดไหม?\\n\\n**คำตอบ:** ผิด\\n\\n**ข้อความทางกฎหมาย:**\\nมาตรา 7 แห่งพระราชบัญญัติรถยนต์ พ.ศ. 2522 ห้ามมิให้ผู้ใดนำรถที่มิได้ติดแผ่นป้ายเลขทะเบียน แผ่นป้ายเครื่องหมายเลขทะเบียน หรือป้ายประจำรถ มาใช้ในทางเดินรถ\\n\\n**ข้อปฏิบัติ:**\\nผู้ขับขี่รถยนต์ทุกคันต้องติดป้ายทะเบียนที่ได้รับอนุญาตจากกรมการขนส่งทางบกไว้ที่ด้านหน้าและด้านหลังของรถ โดยต้องติดให้ชัดเจนและมองเห็นได้ง่าย\\n\\n**คำอธิบายเพิ่มเติม:**\\n- ป้ายทะเบียนเป็นเครื่องหมายสำคัญที่ใช้ในการระบุตัวตนและเจ้าของรถ\\n- การขับรถโดยไม่มีป้ายทะเบียนถือเป็นการผิดกฎหมาย เนื่องจากเจ้าหน้าที่ไม่สามารถระบุตัวตนของผู้ขับขี่หรือเจ้าของรถได้\\n- หากฝ่าฝืนกฎหมายนี้ ผู้ขับขี่อาจถูกปรับไม่เกิน 2,000 บาท และอาจถูกสั่งให้หยุดใช้รถได้', 0, 0, 'No Feed back')\n"
+     ]
+    }
+   ],
+   "source": [
+    "import sqlite3\n",
+    "\n",
+    "# Connect to the SQLite database\n",
+    "conn = sqlite3.connect('feedback.db')\n",
+    "\n",
+    "# Create a cursor object to execute SQL queries\n",
+    "cursor = conn.cursor()\n",
+    "\n",
+    "# Define the SQL query to select all data from the users table\n",
+    "sql_query = 'SELECT * FROM qa_feedback'\n",
+    "\n",
+    "# Execute the SQL query\n",
+    "cursor.execute(sql_query)\n",
+    "\n",
+    "# Fetch all the rows from the result\n",
+    "rows = cursor.fetchall()\n",
+    "\n",
+    "# Print the rows\n",
+    "for row in rows:\n",
+    "    print(row)\n",
+    "\n",
+    "# Close the database connection\n",
+    "conn.close()\n"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "base",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.7"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}