Update app.py

app.py

@@ -1,24 +1,83 @@
-if st.button("Search"):
-    if user_input.strip():
-        with st.spinner("Fetching plant data..."):
-            plant_info = fetch_plant_data(user_input)
-
-        if plant_info:
-            with st.spinner("Generating watering instructions..."):
-                basic_watering = plant_info["watering"] or "No specific instructions available."
-                plant_info["watering"] = refine_watering_instructions(user_input, basic_instructions=basic_watering)
-
-            st.subheader(f"Care Instructions for {plant_info['name']}")
-            st.write(f"**Description:** {plant_info['description']}")
-            st.write(f"**Sun Requirements:** {plant_info['sun_requirements']}")
-            st.write(f"**Watering Instructions:** {plant_info['watering']}")
-            
-            if plant_info["image_url"]:
-                try:
+import streamlit as st
+import requests
+from transformers import pipeline
+
+# Initialize the AI models
+qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad", framework="pt") if no more information is avalable in this link then answer no more information available and erase errors in this app like this streamlit.runtime.media_file_storage.MediaFileStorageError: Error opening '/assets/baren_field_square-4a827e5f09156962937eb100e4484f87e1e788f28a7c9daefe2a9297711a562a.jpg'
+Traceback:
+File "/home/user/app/app.py", line 69, in <module>
+    main()
+File "/home/user/app/app.py", line 64, in main
+    st.image(plant_info["image_url"], caption=plant_info["name"], width=300)
+File "/usr/local/lib/python3.10/site-packages/streamlit/runtime/metrics_util.py", line 409, in wrapped_func
+    result = non_optional_func(*args, **kwargs)
+File "/usr/local/lib/python3.10/site-packages/streamlit/elements/image.py", line 172, in image
+    marshall_images(
+File "/usr/local/lib/python3.10/site-packages/streamlit/elements/lib/image_utils.py", line 438, in marshall_images
+    proto_img.url = image_to_url(
+File "/usr/local/lib/python3.10/site-packages/streamlit/elements/lib/image_utils.py", line 297, in image_to_url
+    url = runtime.get_instance().media_file_mgr.add(image, mimetype, image_id)
+File "/usr/local/lib/python3.10/site-packages/streamlit/runtime/media_file_manager.py", line 226, in add
+    file_id = self._storage.load_and_get_id(
+File "/usr/local/lib/python3.10/site-packages/streamlit/runtime/memory_media_file_storage.py", line 115, in load_and_get_id
+    file_data = self._read_file(path_or_data)
+File "/usr/local/lib/python3.10/site-packages/streamlit/runtime/memory_media_file_storage.py", line 167, in _read_file
+    raise MediaFileStorageError(f"Error opening '{filename}'") from extext_generation_pipeline = pipeline("text-generation", model="gpt2", framework="pt")
+
+# Function to fetch plant data using OpenFarm API
+def fetch_plant_data(plant_name):
+    url = f"https://openfarm.cc/api/v1/crops?filter={plant_name}"
+    try:
+        response = requests.get(url)
+        response.raise_for_status()  # Raise an error for bad status codes
+        data = response.json()
+        if data.get("data"):
+            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", None),
+            }
+        return None
+    except requests.exceptions.RequestException as e:
+        st.error(f"Error fetching data: {e}")
+        return None
+
+# Function to generate watering instructions using AI
+def refine_watering_instructions(plant_name, basic_instructions=None):
+    prompt = (
+        f"Refine these watering instructions for the plant '{plant_name}': '{basic_instructions}'. "
+        "If no specific instructions are available, provide a detailed guideline for watering this plant type."
+    )
+    result = text_generation_pipeline(prompt, max_length=150, num_return_sequences=1)
+    return result[0]["generated_text"]
+
+# Streamlit app
+def main():
+    st.title("🌱 AI-Powered Plant Care Guide")
+
+    # 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", "")
+
+    if st.button("Search"):
+        if user_input.strip():
+            with st.spinner("Fetching plant data..."):
+                plant_info = fetch_plant_data(user_input)
+
+            if plant_info:
+                with st.spinner("Generating watering instructions..."):
+                    basic_watering = plant_info["watering"] or "No specific instructions available."
+                    plant_info["watering"] = refine_watering_instructions(user_input, basic_instructions=basic_watering)
+
+                st.subheader(f"Care Instructions for {plant_info['name']}")
+                st.write(f"**Description:** {plant_info['description']}")
+                st.write(f"**Sun Requirements:** {plant_info['sun_requirements']}")
+                st.write(f"**Watering Instructions:** {plant_info['watering']}")
+                if plant_info["image_url"]:
                     st.image(plant_info["image_url"], caption=plant_info["name"], width=300)
-                except Exception:
-                    st.warning("Could not load the plant image.")
-        else:
-            st.error("No plant information found.")
-    else:
-        st.warning("Please enter a plant name.")
+    
+if __name__ == "__main__":
+    main()