Update app.py

app.py

@@ -7,33 +7,140 @@ import gradio as gr
 import torch
 from gtts import gTTS
 import IPython.display as ipd
-
 device = "cuda" if torch.cuda.is_available() else "cpu"
 translator_ar_to_en = pipeline("translation_ar_to_en", model="Helsinki-NLP/opus-mt-ar-en", device=0 if device == "cuda" else -1)
 translator_en_to_ar = pipeline("translation_en_to_arabic", model="Helsinki-NLP/opus-mt-en-ar", device=0 if device == "cuda" else -1)
-
 # Load the plant identification pipeline from Hugging Face
 pipe = pipeline("image-classification", model="umutbozdag/plant-identity")
-
-# Define the dictionaries for plant care and facts (unchanged)
-plant_care_dict = { ... }
-plant_facts_dict = { ... }
-plant_name_translations = { ... }
-
+plant_care_dict = {
+    "Aloe Vera": "Water every 2-3 weeks, allowing the soil to dry out between waterings. Place in bright, indirect sunlight.",
+    "Bamboo": "Keep the soil moist but not soggy. Place in indirect sunlight, and avoid direct exposure to sunlight.",
+    "Basil": "Water regularly to keep the soil moist. Place in full sun for at least 6 hours a day.",
+    "Boston fern": "Water regularly to maintain humidity. Mist the leaves often and keep in indirect sunlight.",
+    "English ivy": "Water when the top inch of soil is dry. Place in moderate sunlight and maintain humidity.",
+    "Ficus": "Water when the top inch of soil is dry. Place in bright, indirect light, and rotate the plant occasionally.",
+    "Fiddle leaf fig": "Water when the top inch of soil is dry. Place in bright, indirect light, and avoid drafts.",
+    "Lavender": "Water when the soil is dry to the touch. Place in full sun and ensure good drainage.",
+    "Mint": "Water regularly to keep the soil moist. Place in partial shade, and pinch back to encourage bushiness.",
+    "Monstera deliciosa": "Water when the top inch of soil is dry. Place in indirect sunlight and provide support for climbing.",
+    "Orchids": "Water every 1-2 weeks. Provide humidity and place in indirect light.",
+    "Peace Lily": "Water when the top inch of soil is dry. Place in low light and keep out of direct sunlight.",
+    "Philodendron": "Water when the top inch of soil is dry. Place in indirect sunlight, and prune as needed.",
+    "Pothos": "Water when the top inch of soil is dry. Tolerates low light but grows best in bright, indirect light.",
+    "Rosemary": "Water when the top inch of soil is dry. Place in full sun and ensure good drainage.",
+    "Rubber plant": "Water when the top inch of soil is dry. Place in bright, indirect light, and dust leaves regularly.",
+    "Snake plant": "Water every 2-6 weeks, depending on light conditions. Place in low to bright indirect light.",
+    "Spider plant": "Water when the top inch of soil is dry. Place in bright, indirect light, and avoid soggy soil.",
+    "Succulents": "Water every 1-3 weeks, allowing the soil to dry completely between waterings. Place in bright sunlight.",
+    "ZZ plant": "Water when the soil is completely dry, usually every 2-3 weeks. Tolerates low light conditions.",
+    "Jasmine": "Water when the top inch of soil is dry. Place in full sun and prune regularly to encourage growth.",
+    "Rose": "Water regularly, keeping the soil moist but not soggy. Place in full sun and fertilize during the growing season.",
+    "Chamomile": "Water when the top inch of soil is dry. Place in full sun, and harvest flowers regularly.",
+    "Marigold": "Water regularly, especially in dry conditions. Place in full sun and deadhead flowers to promote blooming.",
+    "Daffodil": "Water when the soil is dry to the touch. Place in full sun and avoid overwatering.",
+    "Geranium": "Water when the top inch of soil is dry. Place in full sun and pinch back to promote bushiness.",
+    "Thyme": "Water when the soil is dry. Place in full sun and ensure good drainage.",
+    "Coriander": "Water regularly to keep the soil moist. Place in full sun and harvest leaves often.",
+    "Lemon Balm": "Water when the soil is dry. Place in full sun and prune regularly to encourage growth.",
+    "Sage": "Water when the soil is dry. Place in full sun and provide well-drained soil.",
+    "Peppermint": "Water regularly to keep the soil moist. Place in partial shade and pinch back to promote bushiness.",
+    "Bougainvillea": "Water when the soil is dry. Place in full sun and prune regularly to encourage flowering.",
+    "Desert Rose": "Water when the soil is dry. Place in full sun and ensure good drainage.",
+    "Palm Tree": "Water when the top inch of soil is dry. Place in bright sunlight and provide adequate space for growth.",
+    "Oleander": "Water when the top inch of soil is dry. Place in full sun and fertilize regularly.",
+    "Cactus": "Water every few weeks, allowing the soil to dry completely between waterings. Place in bright sunlight."
+}
+# Plant facts dictionary
+plant_facts_dict = {
+    "Aloe Vera": "Aloe Vera is known for its soothing properties and is commonly used in skincare products. It thrives in bright, indirect sunlight.",
+    "Bamboo": "Bamboo is one of the fastest-growing plants in the world. It prefers moist soil and indirect sunlight.",
+    "Basil": "Basil is a fragrant herb that is often used in cooking. It needs plenty of sunlight and regular watering to thrive.",
+    "Boston fern": "Boston ferns are popular houseplants known for their lush, green fronds. They prefer high humidity and indirect sunlight.",
+    "English ivy": "English ivy is a climbing plant that can adapt to various conditions. It likes well-drained soil and moderate sunlight.",
+    "Ficus": "Ficus plants are versatile and can be grown indoors or outdoors. They prefer bright, indirect light and moderate watering.",
+    "Fiddle leaf fig": "Fiddle leaf figs are known for their large, glossy leaves. They thrive in bright, indirect light and require regular watering.",
+    "Lavender": "Lavender is a fragrant herb often used in aromatherapy. It prefers full sun and well-drained soil.",
+    "Mint": "Mint is a fast-growing herb that can easily spread. It prefers partial shade and moist soil.",
+    "Monstera deliciosa": "Monstera is known for its unique leaf holes. It thrives in indirect sunlight and requires regular watering.",
+    "Orchids": "Orchids are exotic flowers that require special care. They prefer indirect light and humidity.",
+    "Peace Lily": "Peace Lilies are known for their elegant white flowers. They thrive in low light and require regular watering.",
+    "Philodendron": "Philodendrons are popular houseplants with heart-shaped leaves. They prefer indirect sunlight and moderate watering.",
+    "Pothos": "Pothos are resilient plants that can survive in low light. They prefer to dry out between waterings.",
+    "Rosemary": "Rosemary is an aromatic herb that is commonly used in cooking. It prefers full sun and well-drained soil.",
+    "Rubber plant": "Rubber plants are known for their large, glossy leaves. They thrive in bright, indirect light and moderate watering.",
+    "Snake plant": "Snake plants are hardy and can tolerate low light. They require infrequent watering.",
+    "Spider plant": "Spider plants are known for their air-purifying properties. They prefer bright, indirect light and regular watering.",
+    "Succulents": "Succulents are drought-tolerant plants that store water in their leaves. They prefer bright sunlight and well-drained soil.",
+    "ZZ plant": "ZZ plants are known for their glossy leaves and resilience. They thrive in low light and require infrequent watering.",
+    "Jasmine": "Jasmine is a fragrant flower often used in perfumes. It prefers full sun and well-drained soil.",
+    "Rose": "Roses are popular garden plants known for their beautiful flowers. They prefer full sun and regular watering.",
+    "Chamomile": "Chamomile is an aromatic herb used in teas. It prefers full sun and well-drained soil.",
+    "Marigold": "Marigolds are bright flowers that repel pests. They thrive in full sun and require regular watering.",
+    "Daffodil": "Daffodils are spring flowers known for their trumpet shape. They prefer well-drained soil and full sun.",
+    "Geranium": "Geraniums are popular bedding plants. They prefer full sun and regular watering.",
+    "Thyme": "Thyme is an aromatic herb commonly used in cooking. It prefers full sun and well-drained soil.",
+    "Coriander": "Coriander is an herb known for its distinct flavor. It prefers full sun and well-drained soil.",
+    "Lemon Balm": "Lemon balm is a fragrant herb often used in teas. It prefers full sun and regular watering.",
+    "Sage": "Sage is a hardy herb commonly used in cooking. It prefers full sun and well-drained soil.",
+    "Peppermint": "Peppermint is a popular herb known for its refreshing flavor. It prefers partial shade and moist soil.",
+    "Bougainvillea": "Bougainvillea is a colorful flowering plant. It prefers full sun and well-drained soil.",
+    "Desert Rose": "Desert roses are succulent plants known for their beautiful flowers. They prefer full sun and well-drained soil.",
+    "Palm Tree": "Palm trees are iconic tropical plants. They prefer bright sunlight and well-drained soil.",
+    "Oleander": "Oleander is a hardy shrub known for its flowers. It prefers full sun and well-drained soil.",
+    "Cactus": "Cacti are drought-resistant plants that thrive in dry conditions. They prefer bright sunlight and minimal watering."
+}
+plant_name_translations = {
+    "Aloe Vera": "Aloe Vera - صبار الألوفيرا",
+    "Bamboo": "Bamboo - الخيزران",
+    "Basil": "Basil - الريحان",
+    "Boston fern": "Boston Fern - سرخس بوسطن",
+    "English ivy": "English Ivy - اللبلاب الإنجليزي",
+    "Ficus": "Ficus - نبات الفيكس",
+    "Fiddle leaf fig": "Fiddle Leaf Fig - التين ورقي الكمان",
+    "Lavender": "Lavender - الخزامى",
+    "Mint": "Mint - النعناع",
+    "Monstera deliciosa": "Monstera Deliciosa - مونستيرا دليسيوسا",
+    "Orchids": "Orchids - الأوركيد",
+    "Peace Lily": "Peace Lily - زنبق السلام",
+    "Philodendron": "Philodendron - الفيلوديندرون",
+    "Pothos": "Pothos - البوتس",
+    "Rosemary": "Rosemary - إكليل الجبل",
+    "Rubber plant": "Rubber Plant - نبات المطاط",
+    "Snake plant": "Snake Plant - نبات الثعبان",
+    "Spider plant": "Spider Plant - نبات العنكبوت",
+    "Succulents": "Succulents - النباتات العصارية",
+    "ZZ plant": "ZZ Plant - نبات زيزي",
+    "Jasmine": "Jasmine - الياسمين",
+    "Rose": "Rose - الورد",
+    "Chamomile": "Chamomile - البابونج",
+    "Marigold": "Marigold - القطيفة",
+    "Daffodil": "Daffodil - النرجس",
+    "Geranium": "Geranium - إبرة الراعي",
+    "Thyme": "Thyme - الزعتر",
+    "Coriander": "Coriander - الكزبرة",
+    "Lemon Balm": "Lemon Balm - بلسم الليمون",
+    "Sage": "Sage - الميرمية",
+    "Peppermint": "Peppermint - النعناع الفلفلي",
+    "Bougainvillea": "Bougainvillea - الجهنمية",
+    "Desert Rose": "Desert Rose - وردة الصحراء",
+    "Palm Tree": "Palm Tree - النخيل",
+    "Oleander": "Oleander - دفلى",
+    "Cactus": "Cactus - الصبار"}
 def classify_plant_image(image):
     image.save('temp_image.jpg')  # Save the image locally for pipeline processing
     predictions = pipe('temp_image.jpg')
     top_prediction = predictions[0]  # Get the top prediction
     return top_prediction['label']
-
 def text_to_speech(instructions, language='en'):
     tts = gTTS(text=instructions, lang=language)
     tts.save('temp_audio.mp3')  # Save the audio file
     return 'temp_audio.mp3'
-
 def process_image_input(image=None, language='english'):
     plant_name = classify_plant_image(image)  # Identify plant from image
 
+    # Debugging: Log the identified plant name
+    print(f"Identified Plant Name: {plant_name}")
+
     # Get care instructions in English
     care_instructions = plant_care_dict.get(plant_name, f"Care instructions are not available for '{plant_name}'.")
     plant_fact = plant_facts_dict.get(plant_name, f"Facts are not available for '{plant_name}'.")
@@ -49,8 +156,12 @@ def process_image_input(image=None, language='english'):
 
     return plant_name, care_instructions, audio_file
 
+# Main processing function for text input
 def process_text_input(plant_name=None, language='english'):
     if plant_name:
+        # Debugging: Log the requested plant name
+        print(f"Requested Plant Name: {plant_name}")
+
         # Get care instructions in English
         care_instructions = plant_care_dict.get(plant_name, f"Care instructions are not available for '{plant_name}'.")
         plant_fact = plant_facts_dict.get(plant_name, f"Facts are not available for '{plant_name}'.")
@@ -65,43 +176,8 @@ def process_text_input(plant_name=None, language='english'):
         audio_file = text_to_speech(plant_fact, language='ar' if language == 'arabic' else 'en')
 
         return plant_name, care_instructions, audio_file
-
-# Define custom CSS
-css_style = """
-body {
-    background-color: #FFC0CB; /* Baby Pink Background */
-}
-
-.gradio-container {
-    border-radius: 10px;
-    padding: 20px;
-    background-color: white;
-    box-shadow: 2px 2px 8px rgba(0, 0, 0, 0.1);
-}
-"""
-
-# Adding custom JavaScript to dynamically change button and audio styles
-js_script = """
-<script>
-    // Wait for Gradio components to load before applying styles
-    window.onload = function() {
-        // Change button color to green
-        document.querySelectorAll("button").forEach(function(button) {
-            button.style.backgroundColor = '#4CAF50';
-            button.style.color = 'white';
-        });
-
-        // Change audio player background color to green
-        document.querySelectorAll("audio").forEach(function(audio) {
-            audio.style.backgroundColor = '#4CAF50';
-            audio.style.border = '2px solid #45a049';
-        });
-    }
-</script>
-"""
-
-# Creating Gradio app with tabs and custom CSS
-with gr.Blocks(css=css_style) as demo:
+# Creating Gradio app with tabs
+with gr.Blocks() as demo:
     gr.Markdown("## Plant Identification and Care Instructions")
 
     with gr.Tab("Image Input"):
@@ -136,11 +212,8 @@ with gr.Blocks(css=css_style) as demo:
             description="Enter the name of a plant to receive care instructions in your chosen language."
         )
 
-    # Insert JavaScript after Gradio components are rendered
-    gr.HTML(js_script)
-
 # Launch the Gradio app with Tabs
 demo.launch()
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860)