Update app.py

app.py

@@ -13,7 +13,7 @@ genai.configure(api_key=GOOGLE_API_KEY)
 # Use Gemini 1.5 Flash model
 model = genai.GenerativeModel('gemini-1.5-flash-latest')
 
-def authenticate_banknote(image):
+def identify_banknote_denomination(image):
     if image is None:
         return "Error: Please upload or capture a banknote image."
     
@@ -21,40 +21,29 @@ def authenticate_banknote(image):
     image = image.resize((224, 224))
     
     prompt = """
-    You are an image analysis assistant tasked with determining whether a banknote image is authentic or counterfeit and identifying its denomination (worth). 
-    Analyze the image for security features (e.g., security thread, watermark, microtext, color-shifting ink, holograms, printing quality) to determine authenticity. 
-    Identify the denomination based on visual elements like currency symbols, colors, numbers, or portraits.
+    You are an image analysis assistant tasked with identifying the denomination (worth) of a banknote from an image. 
+    Analyze the image for visual elements like currency symbols, colors, numbers, or portraits to determine the denomination.
     Return the response in this format:
     
-    Prediction: [Authentic / Fake]
-    Reason for Authenticity: [Brief explanation]
     Denomination: [Value, e.g., $10, €20]
-    Reason for Denomination: [Brief explanation]
+    Reason: [Brief explanation]
 
     Examples:
-    Image Description: A banknote with a visible security thread that shifts color, a clear watermark visible when held to light, legible microtext under magnification, color-shifting ink, crisp paper texture, sharp printing, and a holographic feature that changes with angle. The note has a green color scheme and a portrait of a historical figure with "$100" printed.
-    Prediction: Authentic
-    Reason for Authenticity: The presence of a color-shifting security thread, clear watermark, legible microtext, color-shifting ink, crisp paper, sharp printing, and holographic features are strong indicators of an authentic banknote.
+    Image Description: A banknote with a green color scheme, a portrait of a historical figure, and "$100" printed prominently.
     Denomination: $100
-    Reason for Denomination: The green color scheme, portrait, and "$100" text are consistent with a U.S. $100 note.
+    Reason: The green color scheme, portrait, and "$100" text are consistent with a U.S. $100 note.
 
-    Image Description: A banknote with no visible security thread, a blurry watermark, illegible microtext, no color-shifting ink, rough paper texture, and blurry printing. The note has a blue color scheme and "€50" printed.
-    Prediction: Fake
-    Reason for Authenticity: The absence of a security thread, blurry watermark, illegible microtext, lack of color-shifting ink, rough paper, and blurry printing suggest a counterfeit banknote.
+    Image Description: A banknote with a blue color scheme, a bridge image, and "€50" printed.
     Denomination: €50
-    Reason for Denomination: The blue color scheme and "€50" text indicate a Euro €50 note, despite its counterfeit nature.
+    Reason: The blue color scheme, bridge image, and "€50" text indicate a Euro €50 note.
 
-    Image Description: A banknote with a security thread, clear watermark, legible microtext, and color-shifting ink. The note has a purple color scheme and "£20" printed.
-    Prediction: Authentic
-    Reason for Authenticity: The presence of a security thread, clear watermark, legible microtext, and color-shifting ink are consistent with an authentic banknote.
+    Image Description: A banknote with a purple color scheme, a specific building, and "£20" printed.
     Denomination: £20
-    Reason for Denomination: The purple color scheme and "£20" text are consistent with a British £20 note.
+    Reason: The purple color scheme, building image, and "£20" text are consistent with a British £20 note.
 
     Analyze the provided image.
-    Prediction:
-    Reason for Authenticity:
     Denomination:
-    Reason for Denomination:
+    Reason:
     """
 
     try:
@@ -65,11 +54,11 @@ def authenticate_banknote(image):
 
 # Define Gradio interface
 iface = gr.Interface(
-    fn=authenticate_banknote,
+    fn=identify_banknote_denomination,
     inputs=gr.Image(type="pil", label="Upload or capture a banknote image", sources=["upload", "webcam"]),
-    outputs=gr.Textbox(label="Banknote Authentication and Denomination Result"),
-    title="Banknote Authenticity and Denomination Predictor",
-    description="Upload or capture a banknote image to determine if it is authentic or counterfeit and identify its denomination using the Gemini API. Set your GOOGLE_API_KEY in the Hugging Face Space settings. Note: This is for demonstration purposes only and not for real-world banking use."
+    outputs=gr.Textbox(label="Banknote Denomination Result"),
+    title="Banknote Denomination Predictor",
+    description="Upload or capture a banknote image to identify its denomination using the Gemini API. Set your GOOGLE_API_KEY in the Hugging Face Space settings. Note: This is for demonstration purposes only and not for real-world banking use."
 )
 
 # Launch the interface