prompt update

mineru_single.py

@@ -181,6 +181,7 @@ class ImageWriter(DataWriter):
 def call_gemini_for_image_description(image_data: bytes) -> str:
     """Convert image bytes to Gemini-compatible format and get description"""
     from google import genai
+    from google.genai import types
     import base64
 
     try:
@@ -193,16 +194,18 @@ def call_gemini_for_image_description(image_data: bytes) -> str:
             contents=[
                 {
                     "parts": [
-                        {"text": """The provided image is a part of a question paper or markscheme. Extract all the necessary information from the image to be able to identify the question. 
-                        For example, if there is an image that contains text like: "Q1 Part A Answer: Life on earth was created by diety..." you should return "Q1 Part A Answer" 
-                        If there is no text on this image, return the description of the image. 20 words max."""},
+                        {"text": """The provided image is a part of a question paper or markscheme. Extract all the necessary information from the image to be able to identify the question.
+To identify the question, we only need the following: question number and question part. Don't include redundant information.
+For example, if image contains text like: "Q1 Part A Answer: Life on earth was created by diety..." you should return just "Q1 Part A Mark Scheme"
+If there is no text on this image, return the description of the image. 20 words max."""},
                         {
                             "inline_data": {
                                 "mime_type": "image/jpeg",
                                 "data": base64.b64encode(image_data).decode('utf-8')
                             }
                         }
-                    ]
+                    ],
+                    config=types.GenerateContentConfig(temperature=0.)
                 }
             ]
         )