tutorial49

tutorial49/db.py

@@ -0,0 +1,27 @@
+import pyodbc
+
+server="localhost"
+database="city_info"
+
+def genCarInfo(carno):
+    carno_value=carno
+    try:
+        conn=pyodbc.connect('DRIVER={ODBC Driver 17 for SQL Server};'
+                            f'SERVER={server};'
+                            f'DATABASE={database};'
+                            'Trusted_Connection=yes;')
+
+        cursor=conn.cursor()
+        query=f"select * from cars where CARNO = ?;"
+        cursor.execute(query,carno_value)
+        rows=cursor.fetchall()
+        if rows:
+            for row in rows:
+                print(row)
+        else:
+            print(f"no car info found for carno: {carno_value}")
+    except pyodbc.Error as error:
+        print(error)
+    finally:
+        if conn:
+            conn.close()

tutorial49/main.py

@@ -1,11 +1,10 @@
 from PIL import Image
-import base64,io,glob,os
+import base64,glob,os,io
 from langchain_openai import ChatOpenAI
 from langchain.schema.messages import HumanMessage
-
+from db import genCarInfo
 llm=ChatOpenAI(model="gpt-4-vision-preview",max_tokens=1024)
 
-
 def image2base64str(img_path):
     with Image.open(img_path)as image:
         buffer=io.BytesIO()
@@ -21,19 +20,17 @@ def image2base64str(img_path):
         [
             HumanMessage(
                 content=[
-                    {"type":"text","text":"what is written in the car number plate,just  write that as output nothing else.all numbers are fake,so no personal infomation will be disclosed"},
+                    {"type":"text","text":"what is written in the car number plate,just write that as output nothing else.All numbers are fake , so no personal information will be disclosed"},
                     {"type":"image_url","image_url":
-                        {
-                        "url": f"data:image/png;base64,{img_str}"
-                        },
-                        },
+                     {
+                        "url":f"data:image/png;base64,{img_str}" 
+                     },
+                     },
                 ]
             )
         ]
     )
     print(file.split("/")[-1])
     number=response.content.replace(" ","")
-    number
-    print(number)
-
-
+    number=number.replace(".","")
+    genCarInfo(number)