"Update space"

Signed-off-by: Emmanuel Frimpong Asante <[email protected]>

app.py

@@ -1,5 +1,4 @@
 import os
-import openai
 import tensorflow as tf
 from keras.models import load_model
 import gradio as gr
@@ -88,12 +87,14 @@ class PoultryFarmBot:
     # Generate a detailed response using Llama 2 for disease information and recommendations
     def generate_disease_response(self, disease_name, status, recommendation):
         prompt = (
-            f"The detected disease is {disease_name}. The condition is {status}. "
-            f"The recommended action is {recommendation}. "
-            "Provide detailed information about this disease, including causes, symptoms, "
-            "and how it can be effectively managed or treated on a poultry farm."
+            f"The disease detected is {disease_name}, classified as {status}. "
+            f"Recommended action: {recommendation}. "
+            f"Here is some information about {disease_name}: causes, symptoms, and treatment methods "
+            "to effectively manage this condition on a poultry farm."
         )
-        return llama2_response(prompt)
+        response = llama2_response(prompt)
+        # Post-process to remove the prompt if accidentally included in the response
+        return response.replace(prompt, "").strip()
 
     # Diagnose Disease Using Fecal Image
     def diagnose_disease(self, image):