Update app.py

app.py

@@ -2,13 +2,20 @@
 # Imports
 # ===============================
 import torch
+import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
 # ===============================
-# Load Model & Tokenizer (CPU)
+# Optional CPU optimizations
 # ===============================
-print("Loading Aksara v1 model (CPU)...")
+torch.set_num_threads(4)  # adjust based on CPU cores
+
+
+# ===============================
+# Load Model & Tokenizer (ONCE)
+# ===============================
+print("Loading Aksara v1 model on CPU...")
 
 tokenizer = AutoTokenizer.from_pretrained(
     "cropinailab/aksara_v1",
@@ -17,16 +24,16 @@ tokenizer = AutoTokenizer.from_pretrained(
 
 model = AutoModelForCausalLM.from_pretrained(
     "cropinailab/aksara_v1",
-    torch_dtype=torch.float32,   # CPU-safe dtype
+    torch_dtype=torch.float32,  # CPU-safe
 )
 
 model.eval()
 
-print("Model loaded successfully on CPU!")
+print("Model loaded successfully!")
 
 
 # ===============================
-# Generation Function
+# Core Generation Function
 # ===============================
 def generate_agri_response(plant, disease):
     prompt = f"""
@@ -48,93 +55,42 @@ Your response MUST follow this structure clearly and must be 100% accurate:
   - Leaves
   - Stems
   - Roots
-  - Fruit (only if the plant actually produces edible fruit; e.g., do NOT mention fruits for potatoes)
+  - Fruit (only if the plant actually produces edible fruit)
   - Tubers/roots if the crop is root-based
-- Do NOT invent symptoms or confuse this disease with others.
 
 ### 3. Safe & Legal Treatment Options
-Provide ONLY safe, standard treatments used by agricultural extension services.
-Include:
 - Copper-based fungicides
 - Mancozeb
 - Chlorothalonil
-- Sulfur (only when relevant)
-- Biological controls (Trichoderma, Bacillus subtilis, Pseudomonas fluorescens)
-- Cultural practices: removing infected leaves, improving airflow, reducing moisture
-
-Rules for treatments:
-- NEVER provide exact dosages, mixing ratios, or spray quantities
-- Do NOT recommend any unsafe or banned chemicals
-- Do NOT claim that neem oil treats fungal/oomycete diseases (only mention it for insects if relevant)
-
-Clarify:
-- Fertilizers DO NOT cure disease; they only improve plant strength and recovery.
+- Biological controls (Trichoderma, Bacillus, Pseudomonas)
+- Cultural practices (airflow, sanitation, moisture control)
+- NO dosages or banned chemicals
 
 ### 4. Prevention
-Include scientifically correct preventive methods:
-- Resistant/tolerant varieties
-- Crop rotation (specify years if relevant)
-- Proper spacing & airflow
-- Mulching and soil moisture control
-- Use of drip irrigation (avoid overhead)
-- Early detection & monitoring
-- Removing volunteer plants and debris
-
-### 5. Nutrient Requirements for This Plant
-Explain the essential nutrients needed for this specific crop:
-- Nitrogen (N)
-- Phosphorus (P)
-- Potassium (K)
-- Calcium (Ca)
-- Magnesium (Mg)
-- Sulfur (S)
-- Micronutrients: Iron (Fe), Zinc (Zn), Boron (B), Manganese (Mn), Copper (Cu), Molybdenum (Mo)
-
-Explain:
-- The role of each nutrient
-- Why it matters for growth, immunity, yield, and stress resistance
+- Resistant varieties
+- Crop rotation
+- Spacing & airflow
+- Drip irrigation
+- Field hygiene
+
+### 5. Nutrient Requirements
+Explain roles of:
+N, P, K, Ca, Mg, S, Fe, Zn, B, Mn, Cu, Mo
 
 ### 6. Fertilizer Recommendations (No Dosages)
-List fertilizers that improve plant nutrition and increase disease resistance (but do NOT treat disease directly):
-
-**Chemical / Mineral Fertilizers**
-- NPK (balanced or crop-specific)
-- Urea (N source)
-- DAP (N + P)
-- MOP or SOP (potassium sources for disease resistance)
-- Rock phosphate (long-term P)
-- Gypsum (Ca + S)
-
-**Organic Fertilizers**
-- Compost
-- Vermicompost
-- Bone meal
-- Seaweed extract (improves immunity & stress tolerance)
-- Panchagavya / Jeevamrut (if culturally relevant)
-
-**Biofertilizers (beneficial microbes)**
-- Azotobacter / Azospirillum (N-fixing)
-- PSB (phosphate-solubilizing bacteria)
-- KMB (potassium-mobilizing bacteria)
-- Trichoderma-enriched compost (suppresses soil-borne pathogens)
-
-For each fertilizer:
-- Explain what nutrient it provides
-- Explain when it is useful (early growth, fruiting, disease recovery, root development)
-- DO NOT provide dosages or exact application rates
+- Chemical
+- Organic
+- Biofertilizers
 
 ### 7. Additional Good Practices
-Include:
-- Irrigation management
-- Soil drainage improvement
+- Irrigation
+- Drainage
 - Tool sanitation
-- Field hygiene
-- Proper storage of harvested produce (if disease affects storage)
 """
 
     inputs = tokenizer(prompt, return_tensors="pt")
 
-    with torch.no_grad():  # important for CPU efficiency
+    with torch.no_grad():
         outputs = model.generate(
             **inputs,
             max_new_tokens=600,
@@ -147,17 +103,65 @@ Include:
 
     full_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-    # Remove echoed prompt
+    # Remove prompt echo safely
     if full_output.startswith(prompt):
-        cleaned = full_output[len(prompt):].strip()
+        response = full_output[len(prompt):].strip()
     else:
-        cleaned = full_output.replace(prompt, "").strip()
+        response = full_output.replace(prompt, "").strip()
+
+    return response
 
-    return cleaned
+
+# ===============================
+# Gradio Interface
+# ===============================
+with gr.Blocks(title="🌱 Agricultural Disease Advisor") as demo:
+    gr.Markdown(
+        """
+        # 🌱 Agricultural Disease Advisor
+        **CPU-based AI assistant for plant disease management**
+        
+        Enter the crop and disease name to receive **safe, scientific, and legal agricultural guidance**.
+        """
+    )
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            plant_input = gr.Textbox(
+                label="🌾 Plant / Crop Name",
+                placeholder="e.g., Potato, Tomato, Rice",
+            )
+
+            disease_input = gr.Textbox(
+                label="🦠 Disease / Issue",
+                placeholder="e.g., Late Blight, Leaf Curl Virus",
+            )
+
+            generate_btn = gr.Button("🔍 Generate Advice", variant="primary")
+
+        with gr.Column(scale=2):
+            output_text = gr.Markdown(
+                label="📋 Agricultural Guidance",
+            )
+
+    generate_btn.click(
+        fn=generate_agri_response,
+        inputs=[plant_input, disease_input],
+        outputs=output_text,
+    )
+
+    gr.Markdown(
+        """
+        ---
+        ⚠️ **Disclaimer:**  
+        This tool provides general agricultural guidance only.  
+        Always consult local agricultural extension services for field-level decisions.
+        """
+    )
 
 
 # ===============================
-# Test
+# Launch
 # ===============================
 if __name__ == "__main__":
-    print(generate_agri_response("Potato", "Late Blight"))
+    demo.launch()