Update app.py

app.py

@@ -13,14 +13,14 @@ from gtts import gTTS
 from googletrans import Translator
 import google.generativeai as genai  # Import Gemini API
 
-# Configure Google Gemini API
+# Configuring Google Gemini API
 GEMINI_API_KEY = os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=GEMINI_API_KEY)
 
-# Load YOLO model for crop disease detection
+# Loading YOLO model for crop disease detection
 yolo_model = YOLO("models/best.pt")
 
-# Initialize conversation history if not set
+# Initializing conversation history if not set
 if "conversation_history" not in st.session_state:
     st.session_state.conversation_history = {}
 
@@ -66,7 +66,7 @@ def generate_gemini_response(disease_list, user_context="", conversation_history
     except Exception as e:
         return f"Error connecting to Gemini API: {str(e)}"
 
-# Perform inference using YOLO
+# Performing inference using YOLO
 def inference(image):
     """Detect crop diseases in the given image."""
     results = yolo_model(image, conf=0.4)
@@ -81,7 +81,7 @@ def inference(image):
 
     return infer, detected_classes, class_names
 
-# Convert text to speech
+# Converting text to chosen language speech
 def text_to_speech(text, language="en"):
     """Convert text to speech using gTTS."""
     try:
@@ -130,14 +130,14 @@ st.subheader("📝 Provide Initial Context or Ask a Question")
 # Generalized example prompts for easier input
 example_prompts = {
     "Select an example...": "",
-    "🌱 General Plant Health Issue": "My plant leaves are wilting and turning yellow. Is this a disease or a nutrient deficiency?",
-    "🛑 Leaf Spots and Discoloration": "I see dark spots on my crop leaves. Could this be a fungal or bacterial infection?",
-    "🍂 Leaves Drying or Curling": "The leaves on my plants are curling and drying up. What could be causing this?",
-    "🧴 Pest or Disease?": "I noticed tiny insects on my plants along with some leaf damage. Could this be a pest problem or a disease?",
-    "💧 Overwatering or Root Rot?": "My plant leaves are turning brown and mushy. Is this due to overwatering or a root infection?",
-    "🌾 Poor Crop Growth": "My crops are growing very slowly and seem weak. Could this be due to soil problems or disease?",
-    "🌡️ Weather and Disease Connection": "It has been raining a lot, and now my plants have mold. Could the weather be causing a fungal disease?",
-    "🌍 Regional Disease Concern": "I'm in a humid area and my crops often get infected. What are common diseases for this climate?",
+    "General Plant Health Issue": "My plant leaves are wilting and turning yellow. Is this a disease or a nutrient deficiency?",
+    "Leaf Spots and Discoloration": "I see dark spots on my crop leaves. Could this be a fungal or bacterial infection?",
+    "Leaves Drying or Curling": "The leaves on my plants are curling and drying up. What could be causing this?",
+    "Pest or Disease?": "I noticed tiny insects on my plants along with some leaf damage. Could this be a pest problem or a disease?",
+    "Overwatering or Root Rot?": "My plant leaves are turning brown and mushy. Is this due to overwatering or a root infection?",
+    "Poor Crop Growth": "My crops are growing very slowly and seem weak. Could this be due to soil problems or disease?",
+    "Weather and Disease Connection": "It has been raining a lot, and now my plants have mold. Could the weather be causing a fungal disease?",
+    "Regional Disease Concern": "I'm in a humid area and my crops often get infected. What are common diseases for this climate?",
 }
 
 # Dropdown menu for selecting an example
@@ -194,13 +194,22 @@ if uploaded_file:
                     st.write("**AI:**", entry["response"])
 
         # Convert diagnosis to speech if enabled
+        from googletrans import Translator
+
         if tts_enabled:
             if st.button("🔊 Listen to Diagnosis"):
                 with st.spinner("Generating audio... 🎵"):
-                    audio_bytes = text_to_speech(diagnosis, language)
+                    # Translate the diagnosis to the target language
+                    translator = Translator()
+                    translated_text = translator.translate(diagnosis, dest=language).text
+                   # Filter out unwanted characters like '#' and '*'
+                    filtered_text = ''.join(c for c in translated_text if c not in ['#', '*'])
+                    # Now process the translated text for TTS
+                    audio_bytes = text_to_speech(filtered_text, language)
                     if audio_bytes:
                         st.audio(audio_bytes, format="audio/mp3")
 
+
     else:
         st.write("❌ No crop disease detected.")