Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 from diffusers import FluxInpaintPipeline
 import random
 import numpy as np
+import google.generativeai as genai
 
 MARKDOWN = """
 # Prompt Canvas🎨
@@ -11,6 +12,34 @@ Thanks to [Black Forest Labs](https://huggingface.co/black-forest-labs) team for
 and a big thanks to [Gothos](https://github.com/Gothos) for taking it to the next level by enabling inpainting with the FLUX.
 """
 
+#Gemini Setup
+genai.configure(api_key = os.environ['Gemini_API'])
+gemini_flash = genai.GenerativeModel(model_name='gemini-1.5-flash-002')
+
+def gemini_predict(prompt):
+    system_message = f"""You are the best text analyser.
+                         You have to analyse a user query and identify what the user wants to change, from a given user query.
+        
+                         Examples:
+                             Query: Change Lipstick colour to blue
+                             Response: Lips
+        
+                             Query: Add a nose stud
+                             Response: Nose
+        
+                             Query: Add a wallpaper to the right wall
+                             Response: Right wall
+        
+                             Query: Change the Sofa's colour to Purple
+                             Response: Sofa
+        
+                        Your response should be in 1 or 2-3 words
+                        Query : {prompt}
+                        """
+    response = gemini_flash.generate_content(system_message)
+    return(response.text)
+
+
 MAX_SEED = np.iinfo(np.int32).max
 DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"