finish gogle gemini implementation

app.py

@@ -78,6 +78,20 @@ def encode_image(pil_image):
 
 #     return image
 
+description_prompt = """
+    I want you to imagine how the technical flat sketch of the garment you see in the picture would look like, and describe it in rich details, in one paragraph.
+    Don't add any additional comment.
+    Specify that the flat sketch is black and white (even if the original garment has a color) and that it doesn't include the person that wear the garment.
+
+
+    The style of the result should look somewhat like the following example:
+    The technical flat sketch of the dress depicts a midi-length, off-the-shoulder design with a smocked bodice and short puff sleeves that have elasticized cuffs.
+    The elastic neckline sits straight across the chest and back, ensuring a secure fit.
+    The bodice transitions into a flowy, tiered skirt with three evenly spaced gathered panels, creating soft volume.
+    Elasticized areas are marked with textured lines, while the gathers and drape is indicated through subtle curved strokes, ensuring clarity in construction details.
+    The flat sketch does NOT include any person and it's only the in black and white, being a technical drawing.
+"""
+
 @spaces.GPU #[uncomment to use ZeroGPU]
 def generate_description_fn(
     image,
@@ -85,44 +99,53 @@ def generate_description_fn(
 ):
     base64_image = encode_image(image)
 
-    client = Groq(
-        api_key=os.environ.get("GROQ_API_KEY"),
+    client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
+    response = client.models.generate_content(
+        model="gemini-2.0-flash",
+        contents=[description_prompt, image]
     )
-
-    chat_completion = client.chat.completions.create(
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": """
-                            I want you to imagine how the technical flat sketch of the garment you see in the picture would look like, and describe it in rich details, in one paragraph.
-                            Don't add any additional comment.
-                            Specify that the flat sketch is black and white (even if the original garment has a color) and that it doesn't include the person that wear the garment.
-
-
-                            The style of the result should look somewhat like the following example:
-                            The technical flat sketch of the dress depicts a midi-length, off-the-shoulder design with a smocked bodice and short puff sleeves that have elasticized cuffs.
-                            The elastic neckline sits straight across the chest and back, ensuring a secure fit.
-                            The bodice transitions into a flowy, tiered skirt with three evenly spaced gathered panels, creating soft volume.
-                            Elasticized areas are marked with textured lines, while the gathers and drape is indicated through subtle curved strokes, ensuring clarity in construction details.
-                            The flat sketch does NOT include any person and it's only the in black and white, being a technical drawing.
-                            """
-                    },
-                    {
-                        "type": "image_url",
-                        "image_url": {
-                            "url": f"data:image/jpeg;base64,{base64_image}",
-                        },
-                    },
-                ],
-            }
-        ],
-        model="llama-3.2-11b-vision-preview",
-    )
-
-    prompt = chat_completion.choices[0].message.content + " In the style of FLTSKC"
+    prompt = response.text + " In the style of FLTSKC"
+
+
+    # client = Groq(
+    #     api_key=os.environ.get("GROQ_API_KEY"),
+    # )
+
+    # chat_completion = client.chat.completions.create(
+    #     messages=[
+    #         {
+    #             "role": "user",
+    #             "content": [
+    #                 {
+    #                     "type": "text",
+    #                     "text": """
+    #                         I want you to imagine how the technical flat sketch of the garment you see in the picture would look like, and describe it in rich details, in one paragraph.
+    #                         Don't add any additional comment.
+    #                         Specify that the flat sketch is black and white (even if the original garment has a color) and that it doesn't include the person that wear the garment.
+
+
+    #                         The style of the result should look somewhat like the following example:
+    #                         The technical flat sketch of the dress depicts a midi-length, off-the-shoulder design with a smocked bodice and short puff sleeves that have elasticized cuffs.
+    #                         The elastic neckline sits straight across the chest and back, ensuring a secure fit.
+    #                         The bodice transitions into a flowy, tiered skirt with three evenly spaced gathered panels, creating soft volume.
+    #                         Elasticized areas are marked with textured lines, while the gathers and drape is indicated through subtle curved strokes, ensuring clarity in construction details.
+    #                         The flat sketch does NOT include any person and it's only the in black and white, being a technical drawing.
+    #                         """
+    #                 },
+    #                 {
+    #                     "type": "image_url",
+    #                     "image_url": {
+    #                         "url": f"data:image/jpeg;base64,{base64_image}",
+    #                     },
+    #                 },
+    #             ],
+    #         }
+    #     ],
+    #     model="llama-3.2-11b-vision-preview",
+    # )
+
+    # prompt = chat_completion.choices[0].message.content + " In the style of FLTSKC"
+    
     control_image = processor(
         image,
         low_threshold=50,