Update app.py

app.py

@@ -1,11 +1,6 @@
 import streamlit as st
 import requests
 from transformers import pipeline
-import logging
-
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
 
 # Cache the AI models to avoid reloading them on every interaction
 @st.cache_resource
@@ -13,7 +8,6 @@ def load_qa_model():
     try:
         return pipeline("question-answering", model="distilbert-base-cased-distilled-squad", framework="pt")
     except Exception as e:
-        logger.error(f"Error loading QA model: {e}")
         st.error(f"Error loading QA model: {e}")
         return None
 
@@ -22,7 +16,6 @@ def load_text_generation_model():
     try:
         return pipeline("text-generation", model="gpt2", framework="pt")
     except Exception as e:
-        logger.error(f"Error loading text generation model: {e}")
         st.error(f"Error loading text generation model: {e}")
         return None
 
@@ -39,25 +32,22 @@ def fetch_plant_data(plant_name):
         response.raise_for_status()  # Raise an error for bad status codes
         data = response.json()
         if data.get("data"):
-            # Return all results for the user to choose from
-            return [
-                {
-                    "name": plant["attributes"].get("name", plant_name),
-                    "description": plant["attributes"].get("description", "No description available."),
-                    "image_url": plant["attributes"].get("main_image_path", None),
-                    "sun_requirements": plant["attributes"].get("sun_requirements", "No information available."),
-                    "watering": plant["attributes"].get("watering", "No specific instructions available."),
-                    "growth_rate": plant["attributes"].get("growth_rate", "No information available."),
-                    "spacing": plant["attributes"].get("spacing", "No information available."),
-                    "sowing_method": plant["attributes"].get("sowing_method", "No information available."),
-                }
-                for plant in data["data"]
-            ]
+            # Get the first result (most likely match)
+            plant_info = data["data"][0]["attributes"]
+            return {
+                "name": plant_info.get("name", plant_name),
+                "description": plant_info.get("description", "No description available."),
+                "image_url": plant_info.get("main_image_path", None),
+                "sun_requirements": plant_info.get("sun_requirements", "No information available."),
+                "watering": plant_info.get("watering", "No specific instructions available."),
+                "growth_rate": plant_info.get("growth_rate", "No information available."),
+                "spacing": plant_info.get("spacing", "No information available."),
+                "sowing_method": plant_info.get("sowing_method", "No information available."),
+            }
         else:
             st.warning(f"No data found for the plant: {plant_name}")
             return None
     except requests.exceptions.RequestException as e:
-        logger.error(f"Error fetching plant data: {e}")
         st.error(f"Error fetching plant data: {e}")
         return None
 
@@ -66,14 +56,12 @@ def refine_watering_instructions(plant_name, basic_instructions=None):
     prompt = (
         f"Provide detailed watering instructions for the plant '{plant_name}'. "
         f"Base your response on the following basic instructions: '{basic_instructions}'. "
-        "Include information on how often to water, the amount of water needed, and any seasonal variations. "
-        "Also, mention signs of overwatering or underwatering."
+        "Include information on how often to water, the amount of water needed, and any seasonal variations."
     )
     try:
-        result = text_generation_pipeline(prompt, max_length=250, num_return_sequences=1)
+        result = text_generation_pipeline(prompt, max_length=200, num_return_sequences=1)
         return result[0]["generated_text"]
     except Exception as e:
-        logger.error(f"Error generating watering instructions: {e}")
         st.error(f"Error generating watering instructions: {e}")
         return "Unable to generate watering instructions at this time."
 
@@ -91,11 +79,6 @@ def is_valid_image_url(url):
 def main():
     st.title("🌱 AI-Powered Plant Care Guide")
 
-    # Sidebar for additional options
-    with st.sidebar:
-        st.header("Options")
-        st.write("Here you can add additional settings or features in the future.")
-
     # Section 1: Search for plant care information
     st.header("Search for Plant Care Information")
     user_input = st.text_input("Enter the name of a plant", "")
@@ -103,50 +86,28 @@ def main():
     if st.button("Search"):
         if user_input.strip():
             with st.spinner("Fetching plant data..."):
-                plant_data = fetch_plant_data(user_input)
+                plant_info = fetch_plant_data(user_input)
 
-            if plant_data:
-                st.success(f"Found {len(plant_data)} result(s) for '{user_input}'!")
+            if plant_info:
+                st.success(f"Found information for {plant_info['name']}!")
                 
-                # Let the user choose the correct plant if there are multiple results
-                if len(plant_data) > 1:
-                    selected_plant = st.selectbox(
-                        "Multiple plants found. Select the correct one:",
-                        options=[plant["name"] for plant in plant_data],
-                        index=0,
-                    )
-                    plant_info = next(plant for plant in plant_data if plant["name"] == selected_plant)
-                else:
-                    plant_info = plant_data[0]
-
                 # Refine or generate AI-based watering instructions
                 with st.spinner("Generating detailed watering instructions..."):
-                    plant_info["watering"] = refine_watering_instructions(plant_info["name"], basic_instructions=plant_info["watering"])
+                    plant_info["watering"] = refine_watering_instructions(user_input, basic_instructions=plant_info["watering"])
 
-                # Display plant information
                 st.subheader(f"Care Instructions for {plant_info['name']}")
-                st.markdown(f"**Description:** {plant_info['description']}")
-                st.markdown(f"**Sun Requirements:** {plant_info['sun_requirements']}")
-                st.markdown(f"**Growth Rate:** {plant_info['growth_rate']}")
-                st.markdown(f"**Spacing:** {plant_info['spacing']}")
-                st.markdown(f"**Sowing Method:** {plant_info['sowing_method']}")
-                
-                # Expandable section for watering instructions
-                with st.expander("View Detailed Watering Instructions"):
-                    st.markdown(f"{plant_info['watering']}")
+                st.write(f"**Description:** {plant_info['description']}")
+                st.write(f"**Sun Requirements:** {plant_info['sun_requirements']}")
+                st.write(f"**Growth Rate:** {plant_info['growth_rate']}")
+                st.write(f"**Spacing:** {plant_info['spacing']}")
+                st.write(f"**Sowing Method:** {plant_info['sowing_method']}")
+                st.write(f"**Watering Instructions:** {plant_info['watering']}")
                 
-                # Feedback mechanism for watering instructions
-                st.write("Was this information helpful?")
-                feedback = st.radio("Feedback", ["Yes", "No"], index=None)
-                if feedback:
-                    st.write(f"Thank you for your feedback! You selected: {feedback}")
-
-                # Display the image if the URL is valid, otherwise use a placeholder
+                # Display the image if the URL is valid
                 if plant_info["image_url"] and is_valid_image_url(plant_info["image_url"]):
                     st.image(plant_info["image_url"], caption=plant_info["name"], width=300)
                 else:
                     st.warning("No valid image available for this plant.")
-                    st.image("https://via.placeholder.com/300x200.png?text=No+Image+Available", caption="Placeholder Image", width=300)
             else:
                 st.warning("No information found for the specified plant. Please try another name.")
         else: