Refactor get_details function to handle multiple image paths

src/app/api/module/product_description.py

@@ -5,12 +5,17 @@ from api.module.utils import  extract_json_from_text
 # from vectorsearch import search , get_detail_df
 import json
 
-def get_details(image_path): 
-    details = azure_ocr(image_path)
-    prompt = base_prompt.format(text = details)
+def get_details(image_path_list): 
+    total_ocr = ""
+    for image_path in image_path_list:
+        details = azure_ocr(image_path)
+        total_ocr += details
+    print(total_ocr)
+    prompt = base_prompt.format(text = total_ocr)
     # print(prompt)
     obj = OpenAIVision()
-    jsontext = obj.get_image_description(image_path,prompt)
+    if len(image_path_list) > 1:
+        jsontext = obj.get_image_description(image_path_list[0],prompt)
     response =  extract_json_from_text(jsontext['choices'][0]['message']['content'])
     return response
 

src/app/api/module/prompts/base.py

@@ -16,11 +16,10 @@ base_prompt = dedent("""
         
         "brand": "sample_brand",
         "mrp": "The price might start with MRP or Rs.",
+        "price": "The price might start with MRP or Rs.",
         "unit": "per pack",
-        "Quantity": 1,  ##num of products visible
+        "Quantity": 1,  ## num of products visible in the image
         "parent_category": "from the above given list",
-        "ingredients": ["ingredient1", "ingredient2", "ingredient3"],
-        "calorie_count": "Would be in numbers",
         "marketed_by": "sample_marketer",
         "manufactured_by": "sample_manufacturer",
         "manufactured_in_country": "Country XYZ",
@@ -28,8 +27,9 @@ base_prompt = dedent("""
         "promotion_on_the_pack": "if any",
         "type_of_product": "give this your understanding",
         "pack_of_or_no_of_units": "No. of Units"
-     
-
+        "description" : "Generate a description of the product"
+        "weight" : "If the weight is mentioned in the Image or OCR Text"
+        
         Analyse data from the above product description to give me the following details in JSON format:
         Only return the output in the required json format.
         """)

src/app2.py

@@ -72,7 +72,7 @@ def sample_fun(voice_input, product_id, progress=gr.Progress()):
 
 with gr.Blocks(theme=gr.themes.Default(primary_hue=gr.themes.colors.red, secondary_hue=gr.themes.colors.pink),title = "Edit Product by Voice") as demo:
     
-    with gr.Tab("Add Your Image"):
+    with gr.Tab("Edit Product by Voice"):
         voice_input = gr.Audio(sources=["microphone"])
         prodcut_id = gr.Textbox(label="Enter Product ID")
         with gr.Row():