Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 from torchvision import transforms
 import warnings
 import sys
+import google.generativeai as genai
 import os
 import contextlib
 from transformers import ViTForImageClassification, pipeline
@@ -39,18 +40,106 @@ transform = transforms.Compose([
 # Load the class names (disease types)
 class_names = ['BacterialBlights', 'Healthy', 'Mosaic', 'RedRot', 'Rust', 'Yellow']
 
-# Load AI response generator (using a local GPT pipeline or OpenAI's GPT-3/4 API)
-ai_pipeline = pipeline("text-generation", model="gpt2", tokenizer="gpt2")
+#Gemini Response 
+def get_response_llm(predicted_label,knowledge_base)
+    prompt = f"Your an helpful assistant who helps farmers know about the sugarcane leaf diseases , precaution, advise etc....Predicted disease label will is given to you  '{predicted_label}' and also {knowledge_base} Provide detailed advice for managing this condition."
+    genai.configure(api_key="YOUR_API_KEY")
+    model = genai.GenerativeModel("gemini-1.5-flash")
+    response = model.generate_content([prompt])
+    return response.text
 
 # Knowledge base for sugarcane diseases (example data from the website)
-knowledge_base = {
-    'BacterialBlights': "Bacterial blights cause water-soaked lesions on leaves, leading to yellowing and withering. To manage, apply copper-based fungicides and ensure proper drainage.",
-    'Mosaic': "Mosaic disease results in streaked and mottled leaves, reducing photosynthesis. Use disease-resistant varieties and control aphids to prevent spread.",
-    'RedRot': "Red rot is identified by reddening and rotting of stalks. Remove infected plants and treat soil with appropriate fungicides.",
-    'Rust': "Rust appears as orange pustules on leaves. Apply systemic fungicides and maintain optimal field conditions to reduce spread.",
-    'Yellow': "Yellowing indicates nutrient deficiencies or initial disease stages. Test soil and provide balanced fertilizers.",
-    'Healthy': "The sugarcane crop is healthy. Continue regular monitoring and good agronomic practices."
-}
+# Comprehensive knowledge base for sugarcane diseases and practices
+knowledge_base = """
+    'BacterialBlights': Bacterial blights are caused by *Xanthomonas albilineans*. 
+    **Symptoms:**
+    - Water-soaked lesions on leaves.
+    - Gradual yellowing and withering of leaves.
+    - Reduction in photosynthesis and stunted growth.
+    
+    **Management:**
+    - Apply copper-based fungicides.
+    - Improve field drainage to avoid waterlogging.
+    - Use disease-free planting material.""",
+    
+    'Mosaic': """Mosaic disease is caused by the Sugarcane mosaic virus (SCMV) and often transmitted by aphids.
+    **Symptoms:**
+    - Mottled appearance on leaves with streaks of yellow and green.
+    - Reduced photosynthetic efficiency.
+    - Decreased cane weight and sugar content.
+    
+    **Management:**
+    - Use resistant sugarcane varieties.
+    - Control aphid populations with insecticides.
+    - Remove and destroy infected plants to prevent spread.""",
+    
+    'RedRot': """Red rot is caused by the fungus *Colletotrichum falcatum*.
+    **Symptoms:**
+    - Red streaks inside the cane with white patches.
+    - Rotting of the stalk, emitting a sour smell.
+    - Drying of leaves and eventual plant death.
+    
+    **Management:**
+    - Plant resistant varieties.
+    - Remove and burn infected plants.
+    - Treat soil with fungicides and practice crop rotation.""",
+    
+    'Rust': """Rust is caused by the fungus *Puccinia melanocephala*.
+    **Symptoms:**
+    - Formation of orange to reddish pustules on leaves.
+    - Premature drying of leaves.
+    - Reduced plant vigor and yield.
+    
+    **Management:**
+    - Apply systemic fungicides (e.g., triazoles).
+    - Ensure proper field hygiene.
+    - Avoid water stress and maintain balanced nutrition.""",
+    
+    'Yellow': """Yellowing can be caused by nutrient deficiencies or disease onset.
+    **Symptoms:**
+    - Yellowing of leaf tips or entire leaves.
+    - Reduced photosynthesis and growth.
+    
+    **Management:**
+    - Conduct soil testing to identify deficiencies.
+    - Apply balanced fertilizers as per soil nutrient status.
+    - Maintain proper irrigation schedules.""",
+    
+    'Smut': """Smut is caused by the fungus *Sporisorium scitamineum*.
+    **Symptoms:**
+    - Formation of whip-like structures at the growing points.
+    - Stunted growth and tiller proliferation.
+    - Reduced sugar content.
+    
+    **Management:**
+    - Plant smut-resistant varieties.
+    - Remove smut-infected plants.
+    - Treat seed sets with fungicides before planting.""",
+    
+    'Healthy': """The sugarcane crop is healthy. 
+    Continue regular monitoring and follow good agronomic practices:
+    - Ensure balanced fertilization.
+    - Maintain proper irrigation schedules.
+    - Monitor for pests and diseases regularly.""",
+
+    'GeneralPractices': """**General Practices for Disease Prevention**
+    - **Field Sanitation:** Remove and destroy crop residues and infected plants to reduce inoculum levels.
+    - **Resistant Varieties:** Cultivate sugarcane varieties that are resistant to specific diseases.
+    - **Seed Treatment:** Use disease-free, certified seed material. Treat seed sets with fungicides before planting.
+    - **Crop Rotation:** Rotate sugarcane with non-host crops to break the disease cycle.
+    - **Optimal Agronomic Practices:** Ensure proper irrigation and drainage. Maintain balanced fertilization and avoid over-application of nitrogen.
+    - **Timely Monitoring and Control:** Inspect fields regularly for symptoms. Apply recommended fungicides or bactericides as soon as symptoms appear.
+    - **Integrated Pest and Disease Management (IPDM):** Combine biological, chemical, and cultural methods to manage diseases sustainably.""",
+
+    'ImpactOfDiseases': """**Impact of Sugarcane Diseases**
+    - **Yield Reduction:** Diseases like red rot and smut can reduce cane yield by 30–60%.
+    - **Quality Degradation:** Affected plants produce less sugar and lower-quality juice.
+    - **Economic Losses:** Increased cost of management and reduced marketable output affect profitability.""",
+
+    'SugarcaneOverview': """Sugarcane is a critical crop globally, providing raw materials for sugar, ethanol, and other byproducts. 
+    However, it is susceptible to various diseases caused by fungi, bacteria, viruses, and environmental factors. 
+    Effective management practices are essential to ensure high yield and quality."""
+
 
 # Update the predict_disease function
 def predict_disease(image):
@@ -65,13 +154,13 @@ def predict_disease(image):
     # Get the predicted label
     predicted_label = class_names[predicted_class.item()]
 
-    # Retrieve response from knowledge base
-    if predicted_label in knowledge_base:
-        detailed_response = knowledge_base[predicted_label]
-    else:
-        # Fallback to AI-generated response
-        prompt = f"The detected sugarcane disease is '{predicted_label}'. Provide detailed advice for managing this condition."
-        detailed_response = ai_pipeline(prompt, max_length=100, num_return_sequences=1, truncation=True)[0]['generated_text']
+    # # Retrieve response from knowledge base
+    # if predicted_label in knowledge_base:
+    #     detailed_response = knowledge_base[predicted_label]
+    # else:
+    #     # Fallback to AI-generated response
+        
+    detailed_response = get_response_llm(predicted_label,knowledge_base)
 
     # Create a styled HTML output
     output_message = f"""