Update app.py

app.py

@@ -1,10 +1,375 @@
+import os
+import gradio as gr
+import re
+from fastai.vision.all import *
+from groq import Groq
+from PIL import Image
+
+# Load the trained model for grapevine disease classification
+learn = load_learner('export.pkl')  # Assumes you have a trained model for grapevine diseases
+labels = learn.dls.vocab
+
+# Initialize Groq client
+client = Groq(
+    api_key=os.environ.get("GROQ_API_KEY"),
+)
+
+# Your existing functions remain the same
+def clean_disease_name(name):
+    """Clean disease name by removing numbers and special characters, and fix formatting"""
+    # Remove numbers and dots at the beginning
+    cleaned = re.sub(r'^\d+\.', '', name)
+    # Replace underscores with spaces
+    cleaned = cleaned.replace('_', ' ')
+    # Remove any remaining special characters
+    cleaned = re.sub(r'[^\w\s]', '', cleaned)
+    # Fix spacing
+    cleaned = ' '.join(cleaned.split())
+    return cleaned
+
+def format_disease_info(raw_info):
+    """Improve the formatting of disease information"""
+    # Add proper line breaks between sections and ensure consistent heading levels
+    formatted = raw_info
+   
+    # Replace markdown headings with HTML headings for better control
+    formatted = re.sub(r'#+\s+(.*)', r'<h3>\1</h3>', formatted)
+   
+    # Add paragraph tags for better spacing
+    formatted = re.sub(r'\n\*\s+(.*)', r'<p>• \1</p>', formatted)
+    formatted = re.sub(r'\n([^<\n].*)', r'<p>\1</p>', formatted)
+   
+    # Remove any duplicate paragraph tags
+    formatted = formatted.replace('<p><p>', '<p>')
+    formatted = formatted.replace('</p></p>', '</p>')
+   
+    return formatted
+
+def get_disease_treatment(disease_name, language="swahili"):
+    """Get direct treatment measures for the disease"""
+    clean_name = clean_disease_name(disease_name)
+   
+    # Return healthy message if the prediction is healthy
+    if clean_name.lower() == "healthy" or "healthy" in clean_name.lower():
+        if language.lower() == "swahili":
+            return """
+            ### Hali ya Jani: Afya nzuri
+           
+            Jani hili liko katika hali nzuri ya afya. Hakuna dalili za magonjwa zinazoonekana. Endelea kutunza mizabibu yako vizuri kwa:
+           
+            • Kumwagilia maji ipasavyo
+            • Kuhakikisha rutuba ya udongo
+            • Kupunguza matawi kwa usahihi
+            • Kufuatilia mara kwa mara kwa dalili za magonjwa
+            • Kudumisha hewa nzuri kuzunguka mmea
+            """
+        else:
+            return """
+            ### Leaf Status: Healthy
+           
+            This leaf is in a healthy state. No signs of disease are present. Continue to maintain your vines by:
+           
+            • Proper irrigation
+            • Ensuring soil fertility
+            • Appropriate pruning
+            • Regular monitoring for disease symptoms
+            • Maintaining good air circulation around the plant
+            """
+   
+    prompt = f"""
+    Provide a concise, practical list of DIRECT treatment measures for {clean_name} disease in grapevines.
+   
+    Format your response as a bulleted list with 5-7 specific actions a vineyard manager should take IMMEDIATELY to treat the disease.
+   
+    For each treatment measure:
+    1. Start with an action verb
+    2. Be specific about products/techniques to use
+    3. Include application rates/timing when relevant
+   
+    The response should be in {"Swahili" if language.lower() == "swahili" else "English"}.
+   
+    Begin with a short title: "HATUA ZA MATIBABU:" (if Swahili) or "TREATMENT MEASURES:" (if English)
+    """
+   
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            model="llama-3.3-70b-versatile",
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        error_msg = "Hitilafu kupata taarifa za matibabu" if language.lower() == "swahili" else "Error fetching treatment information"
+        return f"{error_msg}: {str(e)}"
+
+def get_disease_info(disease_name, language="swahili"):
+    """Get detailed information about a grapevine disease using Groq API"""
+    clean_name = clean_disease_name(disease_name)
+   
+    # Return healthy message if the prediction is healthy
+    if clean_name.lower() == "healthy" or "healthy" in clean_name.lower():
+        if language.lower() == "swahili":
+            return """
+            # Hali ya Jani: Afya nzuri
+           
+            Jani hili liko katika hali nzuri ya afya. Hakuna dalili za magonjwa zinazoonekana. Hii ni ishara nzuri kwamba:
+           
+            - Mizabibu yako inapata virutubisho vya kutosha
+            - Umwagiliaji wa maji uko katika kiasi cha kutosha
+            - Hali ya hewa katika shamba lako ni nzuri
+            - Hakuna mashambulizi ya wadudu waharibifu
+           
+            Endelea kudumisha utunzaji mzuri wa mizabibu yako ili kuhakikisha uzalishaji bora wa zabibu.
+            """
+        else:
+            return """
+            # Leaf Status: Healthy
+           
+            This leaf is in a healthy state. No signs of disease are present. This is a good indication that:
+           
+            - Your vines are receiving adequate nutrition
+            - Irrigation levels are appropriate
+            - Climate conditions in your vineyard are favorable
+            - There are no pest invasions
+           
+            Continue to maintain good vineyard practices to ensure optimal grape production.
+            """
+   
+    lang = "SWAHILI" if language.lower() == "swahili" else "ENGLISH"
+   
+    prompt = f"""
+    Provide detailed information IN {lang} about {clean_name} disease in grapevines.
+   
+    Structure your response like this:
+    1. A brief introduction about what this disease is (1-2 sentences)
+    2. Main symptoms and how to identify the disease (short paragraph)
+    3. FOCUS EXTENSIVELY on the treatment methods and prevention strategies (this should be the longest section)
+    4. Conclusion with practical advice for vineyard managers
+   
+    Format your response in markdown for better readability.
+    Make sure the entire response is in {lang}.
+    """
+   
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            model="llama-3.3-70b-versatile",
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        error_msg = "Hitilafu kupata taarifa" if language.lower() == "swahili" else "Error fetching information"
+        return f"{error_msg}: {str(e)}"
+
+def predict_and_get_info(img, language="swahili"):
+    """Predict grapevine disease and get detailed information"""
+    # Process the image
+    img = PILImage.create(img)
+   
+    # Get prediction
+    pred, pred_idx, probs = learn.predict(img)
+   
+    # Get top 5 predictions (or all if less than 5)
+    num_classes = min(5, len(labels))
+    top_indices = probs.argsort(descending=True)[:num_classes]
+    top_probs = probs[top_indices]
+    top_labels = [labels[i] for i in top_indices]
+   
+    # Format as dictionary with cleaned names for display
+    prediction_results = {clean_disease_name(top_labels[i]): float(top_probs[i]) for i in range(num_classes)}
+   
+    # Get top prediction (original format for info retrieval)
+    top_disease = str(pred)
+    # Also keep a clean version for display
+    clean_top_disease = clean_disease_name(top_disease)
+   
+    # Check if healthy
+    is_healthy = clean_top_disease.lower() == "healthy" or "healthy" in clean_top_disease.lower()
+   
+    # Get detailed information about the top predicted disease
+    disease_info = get_disease_info(top_disease, language)
+    formatted_info = format_disease_info(disease_info)
+   
+    # Get direct treatment measures
+    treatment_measures = get_disease_treatment(top_disease, language)
+    formatted_treatment = format_disease_info(treatment_measures)
+   
+    # Create title text based on language and health status
+    if language.lower() == "swahili":
+        title_text = "Jani Zuri na Afya Njema" if is_healthy else f"Ugonjwa: {clean_top_disease}"
+        advice_title = "Ushauri wa Haraka kwa Wakulima"
+        advice_text = "Kumbuka: Fuata maelekezo ya kitaalamu na tumia dawa zilizoidhinishwa tu. Wasiliana na mtaalamu wa kilimo kwa ushauri zaidi."
+        treatment_title = "Hatua za Matibabu"
+    else:
+        title_text = "Healthy Leaf" if is_healthy else f"Disease: {clean_top_disease}"
+        advice_title = "Quick Advice for Vineyard Managers"
+        advice_text = "Remember: Follow professional instructions and use only approved treatments. Contact an agricultural expert for more advice."
+        treatment_title = "Treatment Measures"
+   
+    # Create well-formatted display with improved fonts and styling
+    custom_css = """
+    <style>
+    .disease-container {
+        font-family: 'Nunito', 'Segoe UI', sans-serif;
+        padding: 20px;
+        max-width: 100%;
+        background-color: #ffffff;
+        border-radius: 12px;
+        box-shadow: 0 4px 12px rgba(0, 0, 0, 0.08);
+    }
+    .disease-title {
+        color: #2a633d;
+        font-size: 28px;
+        font-weight: 700;
+        margin-bottom: 20px;
+        padding-bottom: 10px;
+        border-bottom: 2px solid #e0e0e0;
+    }
+    .info-container {
+        line-height: 1.8;
+        font-size: 16px;
+        color: #333333;
+    }
+    .info-container h3 {
+        color: #2a633d;
+        font-size: 22px;
+        font-weight: 600;
+        margin-top: 25px;
+        margin-bottom: 15px;
+        padding-bottom: 8px;
+        border-bottom: 1px solid #e0e0e0;
+    }
+    .info-container p {
+        margin-bottom: 15px;
+        text-align: justify;
+    }
+    .treatment-section {
+        background-color: #f1f8e9;
+        padding: 20px;
+        border-radius: 8px;
+        margin: 25px 0;
+        border-left: 5px solid #558b2f;
+    }
+    .treatment-section h3 {
+        color: #33691e;
+        border-bottom: none;
+        margin-top: 0;
+    }
+    .direct-treatment {
+        background-color: #e8f5e9;
+        padding: 20px;
+        border-radius: 8px;
+        margin: 25px 0;
+        border-left: 5px solid #2e7d32;
+    }
+    .direct-treatment h3 {
+        color: #2e7d32;
+        border-bottom: none;
+        margin-top: 0;
+    }
+    .healthy-state {
+        background-color: #e8f5e9;
+        color: #2e7d32;
+        padding: 15px;
+        border-radius: 8px;
+        text-align: center;
+        font-weight: bold;
+        font-size: 18px;
+        margin: 15px 0;
+    }
+    /* Ensure good readability for text */
+    body {
+        font-feature-settings: "kern", "liga", "clig", "calt";
+        text-rendering: optimizeLegibility;
+    }
+    </style>
+    """
+   
+    # Healthy banner HTML if needed
+    healthy_banner = ""
+    if is_healthy:
+        if language.lower() == "swahili":
+            healthy_banner = '<div class="healthy-state">✓ Jani hili liko katika hali nzuri ya afya</div>'
+        else:
+            healthy_banner = '<div class="healthy-state">✓ This leaf is in a healthy state</div>'
+   
+    combined_info = f"""
+    {custom_css}
+    <div class="disease-container">
+        <h2 class="disease-title">{title_text}</h2>
+        {healthy_banner}
+       
+        <div class="info-container">
+            {formatted_info}
+        </div>
+       
+        <div class="direct-treatment">
+            <h3>{treatment_title}</h3>
+            {formatted_treatment}
+        </div>
+       
+        <div class="treatment-section">
+            <h3>{advice_title}</h3>
+            <p>{advice_text}</p>
+        </div>
+    </div>
+    """
+   
+    return prediction_results, combined_info, clean_top_disease
+
+def follow_up_question(question, disease_name, language="swahili"):
+    """Allow vineyard managers to ask follow-up questions about the identified disease"""
+    if not question.strip() or not disease_name:
+        return "Tafadhali tambua ugonjwa wa mizabibu kwanza na uulize swali mahususi kuuhusu." if language.lower() == "swahili" else "Please identify a grapevine disease first and ask a specific question about it."
+   
+    lang = "SWAHILI" if language.lower() == "swahili" else "ENGLISH"
+   
+    prompt = f"""
+    The vineyard manager is asking about {disease_name} disease: "{question}"
+   
+    Provide a detailed answer IN {lang}, focusing on accurate plant pathology information.
+    If the question relates to treatment options, preventive measures, or climate change impacts, emphasize those aspects in your response.
+   
+    IMPORTANT:
+    - Do not repeat basic introductory information about the disease that would have already been provided
+    - Directly answer the specific question asked
+    - The entire response must be in {lang}
+    - Focus on practical advice whenever possible
+   
+    Format your response in markdown for better readability.
+    """
+   
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            model="llama-3.3-70b-versatile",
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        error_msg = "Hitilafu kupata taarifa" if language.lower() == "swahili" else "Error fetching information"
+        return f"{error_msg}: {str(e)}"
+
+# Enhanced CSS for styling the improved language toggle
 toggle_css = """
 <style>
 /* Main container for language toggle */
 .language-toggle-container {
     display: flex;
-    flex-direction: column;
-    align-items: flex-end;
+    align-items: center;
+    justify-content: flex-end;
     margin-bottom: 20px;
     padding: 10px;
     background-color: rgba(255, 255, 255, 0.9);
@@ -12,15 +377,6 @@ toggle_css = """
     box-shadow: 0 2px 8px rgba(0, 0, 0, 0.05);
 }
 
-/* Toggle title styling */
-.toggle-title {
-    font-size: 16px;
-    font-weight: 600;
-    margin-bottom: 5px;
-    color: #2a633d;
-    align-self: center;
-}
-
 /* Toggle label styling */
 .toggle-label {
     font-size: 16px;
@@ -61,6 +417,9 @@ toggle_css = """
     font-size: 14px !important;
     transition: all 0.2s ease !important;
     color: #555 !important;
+    display: flex !important;
+    align-items: center !important;
+    justify-content: center !important;
 }
 
 .language-switcher .gr-radio input:checked + label {
@@ -92,7 +451,6 @@ toggle_css = """
 @media (max-width: 768px) {
     .language-toggle-container {
         justify-content: center;
-        align-items: center;
     }
     
     .language-switcher .gr-radio label {
@@ -123,7 +481,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=toggle_css) as app:
             "submit_btn": "Wasilisha Swali",
             "clear_btn": "Futa Mazungumzo",
             "toggle_label": "Lugha:",
-            "toggle_title": "Swahili/English",
             "error_no_image": "Tafadhali pakia picha",
             "error_processing": "Hitilafu katika kuchakata picha"
         },
@@ -140,7 +497,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=toggle_css) as app:
             "submit_btn": "Submit Question",
             "clear_btn": "Clear Conversation",
             "toggle_label": "Language:",
-            "toggle_title": "Swahili/English",
             "error_no_image": "Please upload an image",
             "error_processing": "Error processing image"
         }
@@ -153,31 +509,19 @@ with gr.Blocks(theme=gr.themes.Soft(), css=toggle_css) as app:
     # Store the current disease for context
     current_disease = gr.State("")
    
-    # Add improved language toggle at the top with title
+    # Add improved language toggle at the top
     with gr.Row(elem_classes="language-toggle-container"):
-        # Add the title above the toggle
-        toggle_title = gr.Markdown(text_dict["swahili"]["toggle_title"], elem_classes="toggle-title language-transition")
+        toggle_label = gr.Markdown(text_dict["swahili"]["toggle_label"], elem_classes="toggle-label language-transition")
         
-        with gr.Row():
-            toggle_label = gr.Markdown(text_dict["swahili"]["toggle_label"], elem_classes="toggle-label language-transition")
-            
-            # Custom HTML for flag icons
-            flag_html = gr.HTML("""
-            <div style="display:none">
-                <img id="sw-flag" src="https://flagcdn.com/w20/tz.png" alt="Swahili Flag">
-                <img id="en-flag" src="https://flagcdn.com/w20/gb.png" alt="English Flag">
-            </div>
-            """)
-            
-            with gr.Column(elem_classes="language-switcher", scale=1):
-                language_toggle = gr.Radio(
-                    ["Swahili", "English"], 
-                    value="Swahili", 
-                    label="", 
-                    interactive=True,
-                    elem_id="language-toggle"
-                )
-
+        with gr.Column(elem_classes="language-switcher", scale=1):
+            language_toggle = gr.Radio(
+                ["Swahili", "English"], 
+                value="Swahili", 
+                label="", 
+                interactive=True,
+                elem_id="language-toggle"
+            )
+   
     # Main identification section
     with gr.Row():
         with gr.Column(scale=1):
@@ -208,14 +552,14 @@ with gr.Blocks(theme=gr.themes.Soft(), css=toggle_css) as app:
         follow_up_btn = gr.Button(text_dict["swahili"]["submit_btn"], variant="primary", elem_classes="language-transition")
         clear_btn = gr.Button(text_dict["swahili"]["clear_btn"], elem_classes="language-transition")
    
-    # Enhanced function to switch language based on toggle - updated to include the new title
+    # Enhanced function to switch language based on toggle
     def switch_language(selected_language, current_lang):
         # Map the radio value to our language code
         new_lang = "swahili" if selected_language == "Swahili" else "english"
         
         # If language hasn't changed, don't update anything
         if new_lang == current_lang:
-            return [current_lang] + [gr.update() for _ in range(12)]
+            return [current_lang] + [gr.update() for _ in range(11)]
         
         # Debug message to verify the language switch
         print(f"Switching language from {current_lang} to {new_lang}")
@@ -241,8 +585,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=toggle_css) as app:
             gr.update(label=text_dict[new_lang]["question_label"], placeholder=text_dict[new_lang]["question_placeholder"]),
             text_dict[new_lang]["submit_btn"],
             text_dict[new_lang]["clear_btn"],
-            text_dict[new_lang]["toggle_label"],
-            text_dict[new_lang]["toggle_title"]
+            text_dict[new_lang]["toggle_label"]
         )
    
     # Set up event handlers
@@ -282,7 +625,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=toggle_css) as app:
     def clear_conversation_history():
         return ""
    
-    # Connect the language toggle with enhanced functionality - updated to include the new title
+    # Connect the language toggle with enhanced functionality
     language_toggle.change(
         switch_language,
         inputs=[language_toggle, current_language],
@@ -298,8 +641,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=toggle_css) as app:
             follow_up_input,
             follow_up_btn,
             clear_btn,
-            toggle_label,
-            toggle_title
+            toggle_label
         ]
     )