changed getReg from 2 LLM calls to 1 LLM call

app.py

@@ -8,7 +8,7 @@ import requests
 from io import BytesIO
 import base64
 import re
-
+import json
 
 # Load environment variables from .env
 load_dotenv()
@@ -28,6 +28,11 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 
 #login(token=HF_TOKEN)
 
+#I'm using a virtual environment for this locally
+#python -m venv getReg
+#source getReg/Scripts/activate  (for windows)
+#pip install -r requirements.txt
+#python app.py
 
 
 
@@ -56,8 +61,9 @@ def load_image(image_input):
         raise TypeError("Expected image input to be a string")
 
 
-def product_identification_response(image_path=r"C:\Users\JoeJo\Downloads\XyAaqBEtYtb8YffjKZ68Gb.jpg"):
-    
+def product_identification_response(image_path=r"C:\Users\JoeJo\Downloads\batty car front.jpg"):
+    #"C:\Users\JoeJo\Downloads\batty car front.jpg"
+
     # Authenticate
     genai.configure(api_key=os.environ.get("GENAI_API_KEY"))
     
@@ -66,13 +72,30 @@ def product_identification_response(image_path=r"C:\Users\JoeJo\Downloads\XyAaqB
     
     image = load_image(image_path)
     
+    schema = {
+        "type": "object",
+        "properties": {
+            "registration_number": {"type": "string", "description": "registration number of the vehicle in this image"}
+            },
+            "required": ["registration_number"]
+        }
     
     # Ask Gemini
     response = model.generate_content(
-        ["What is the registration number of the vehicle in this image", image]
+        contents=["What is the registration number of the vehicle in this image", image],
+        generation_config={
+            "response_mime_type": "application/json",
+            "response_schema": schema
+        }
     )
     
     print(f"gemini-1.5-flash answer is: {response.text}")
+
+
+    data = json.loads(response.text)
+    print(f"data after pushing response into JSON is: {data}")
+    return data
+
     
     prompt = f"""Your task is to returned structured JSON of product and condition in the following format: {{ "product": "the identity of the product", "condition": "the condition of the product"}}.
     The condition of the product must be one of the following: (*) New, (*) Like New, (*) Good or (*) Poor.
@@ -107,7 +130,7 @@ def product_identification_response(image_path=r"C:\Users\JoeJo\Downloads\XyAaqB
 demo = gr.Interface(
     fn=product_identification_response,
     inputs="text",
-    outputs="text",
+    outputs="json",
     title="identify registration number",
     description="finds info about a product"
 )