Create app.py

app.py

@@ -0,0 +1,173 @@
+# ===============================
+# Imports
+# ===============================
+import torch
+import spaces
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+
+# ===============================
+# Load Model & Tokenizer (ONCE)
+# ===============================
+print("Loading Aksara v1 model...")
+
+tokenizer = AutoTokenizer.from_pretrained(
+    "cropinailab/aksara_v1",
+    use_fast=True,
+)
+
+model = AutoModelForCausalLM.from_pretrained(
+    "cropinailab/aksara_v1",
+    torch_dtype=torch.bfloat16,
+    device_map="cuda",          # ensures GPU usage on Spaces
+)
+
+model.eval()
+
+print("Model loaded successfully!")
+
+
+# ===============================
+# Generation Function (GPU)
+# ===============================
+@spaces.GPU
+def generate_agri_response(plant, disease):
+    prompt = f"""
+You are an agricultural expert specializing in plant pathology, crop nutrition, and safe farm management.
+Your job is to provide accurate, scientifically correct, and legally safe advice.
+
+Plant: {plant}
+Issue: {disease}
+
+Your response MUST follow this structure clearly and must be 100% accurate:
+
+### 1. About the Disease
+- Explain what the disease is and identify the correct pathogen type (fungus, bacteria, virus, pest, oomycete, etc.)
+- Describe how it spreads (only scientifically correct modes of spread)
+- Avoid any incorrect or exaggerated claims
+
+### 2. Symptoms
+- Describe accurate symptoms on each relevant plant part:
+  - Leaves
+  - Stems
+  - Roots
+  - Fruit (only if the plant actually produces edible fruit; e.g., do NOT mention fruits for potatoes)
+  - 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.
+
+### 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
+
+### 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
+
+### 7. Additional Good Practices
+Include:
+- Irrigation management
+- Soil drainage improvement
+- Tool sanitation
+- Field hygiene
+- Proper storage of harvested produce (if disease affects storage)
+"""
+
+    # Tokenize
+    inputs = tokenizer(
+        prompt,
+        return_tensors="pt",
+    ).to(model.device)
+
+    # Generate
+    outputs = model.generate(
+        **inputs,
+        max_new_tokens=600,
+        temperature=0.7,
+        top_p=0.9,
+        repetition_penalty=1.15,
+        do_sample=True,
+        pad_token_id=tokenizer.eos_token_id,
+    )
+
+    full_output = tokenizer.decode(
+        outputs[0],
+        skip_special_tokens=True,
+    )
+
+    # Remove echoed prompt safely
+    if full_output.startswith(prompt):
+        cleaned = full_output[len(prompt):].strip()
+    else:
+        cleaned = full_output.replace(prompt, "").strip()
+
+    return cleaned
+
+
+# ===============================
+# Local Test (optional)
+# ===============================
+if __name__ == "__main__":
+    print(generate_agri_response("Potato", "Late Blight"))