Update app.py

app.py

@@ -8,6 +8,7 @@ import os
 genai.configure(api_key=os.getenv("genai"))
 oapi_key= os.getenv("openai")
 
+
 def encode_image(image_file):
     with open(image_file.name, "rb") as img_file:
         return base64.b64encode(img_file.read()).decode('utf-8')
@@ -123,11 +124,8 @@ def send_to_openai(image_file):
         "threshold": "BLOCK_MEDIUM_AND_ABOVE"
       },
     ]
-    
-    model = genai.GenerativeModel(model_name="gemini-1.0-pro",
-                                  generation_config=generation_config,
-                                  safety_settings=safety_settings)
-    
+    model = "gemini-1.0-pro"
+
     prompt_parts = [
       "You are a Rice Paddy Agricultural Scientist , You will be given the state of the paddy(a picture description ) and you will have to advice on the correct remedyInput - Sheath Yes Meaning -  The Farmer has shown you a picture of Sheath part of the paddy and it is suffering from Sheath Rot , advice the farmer on how to fix thisInput - Sheath No Meaning - The Farmer has shown you a picture of Sheath part of the paddy and it is healthy ,congratulate and advice farmer on how to maintain itInput - Leaf Yes No No Meaning - The Farmer has shown you a picture of Leaf part of the paddy and it is healthy , congratulate the farmer and advice the farmer on how to maintain it Input - Leaf No Yes No Meaning - The Farmer has shown you a picture of Leaf part of the paddy and it has brown spots disease  ,  advice the farmer on how to remedy it Input - Leaf No Yes YesMeaning - The Farmer has shown you a picture of Leaf part of the paddy and it has brown spots disease and Blight Disease  ,  advice the farmer on how to remedy it Input - Leaf No No YesMeaning - The Farmer has shown you a picture of Leaf part of the paddy and it has Blight disease  ,  advice the farmer on how to remedy it Input - Leaf Yes No Yes Meaning - The Farmer has shown you a picture of Leaf part of the paddy and it looks healthy but it has yellowish  edges that might indicate blight ,  advice the farmer on how to remedy it Input - Leaf Yes Yes No Meaning - The Farmer has shown you a picture of Leaf part of the paddy and it has brown spots disease but also is quite green so it might be healthy and just got the brown spot disease  ,  advice the farmer on how to remedy it",
       "input: Sheath Yes",
@@ -161,12 +159,19 @@ def send_to_openai(image_file):
         new_prompt_parts.append(f"output: {output_key}")
     gemini = genai.GenerativeModel(model_name=model)
 
-    response = gemini.generate_content(
-        contents,
-        generation_config=generation_config,
-        safety_settings=safety_settings,
-        stream=False)
-    return(response.text)
+    model = genai.GenerativeModel(model_name="gemini-1.0-pro",
+                                  generation_config=generation_config,
+                                  safety_settings=safety_settings)
+    response = model.generate_content(prompt_parts)
+    input_text=response.text
+    output_index = input_text.find("output:")
+
+    if output_index != -1:
+        # Extract the text after "output:"
+        output_text = input_text[output_index + len("output:"):].strip()
+
+        # Print the extracted text
+        return output_text