Update app.py

app.py

@@ -70,7 +70,8 @@ def process_text(text_input, unit):
     elif text_input and endpoints == "GOOGLE":
         genai.configure(api_key=GOOGLE_API_KEY)
         model = genai.GenerativeModel(MODEL),
-        return response = model.generate_content(SYS_PROMPT + text_input).text
+        response = model.generate_content(f" You are a experienced {unit} doctor." + SYS_PROMPT + f"Could you solve {text_input}?")
+        return response.text
     return ""
 
 def encode_image_to_base64(image_input):
@@ -105,8 +106,8 @@ def process_image(image_input, unit):
     elif image_input is not None and ENDPOINTS == "GOOGLE":
         genai.configure(api_key=GOOGLE_API_KEY)
         model = genai.GenerativeModel(MODEL),
-        return response = model.generate_content(image_input).text
-
+        response = model.generate_content([f" You are a experienced {unit} doctor." + SYS_PROMPT + "Help me understand what is in this picture and analysis.", image_input])
+        return response.text
 
 def main(text_input="", image_input=None, unit=""):
     if text_input and image_input is None: