Update app.py

app.py

@@ -55,14 +55,14 @@ def predict_disease(image):
     # Get the predicted label
     predicted_label = class_names[predicted_class.item()]
 
-    # Generate AI response based on the detected disease
+    # Generate a detailed response for the detected disease
     prompt = f"The detected sugarcane disease is '{predicted_label}'. Provide detailed advice for managing this condition."
-    ai_response = ai_pipeline(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
-    
-    # Remove the prompt text if it appears in the AI response
-    if ai_response.startswith(prompt):
-        ai_response = ai_response[len(prompt):].strip()
+    ai_response = ai_pipeline(prompt, max_length=100, num_return_sequences=1, truncation=True)[0]['generated_text']
     
+    # Post-process the AI response to ensure it ends with a complete sentence
+    if not ai_response.endswith(('.', '!', '?')):
+        ai_response = ai_response.rsplit('.', 1)[0] + '.'
+
     # Create a styled HTML output
     output_message = f"""
     <div style='font-size: 18px; color: #4CAF50; font-weight: bold;'>
@@ -73,7 +73,7 @@ def predict_disease(image):
     if predicted_label != "Healthy":
         output_message += f"""
         <p style='font-size: 16px; color: #757575;'>
-            <strong>AI Response:</strong> {ai_response}
+            {ai_response}
         </p>
         """
     else: