Update app.py

app.py

@@ -1,69 +1,24 @@
-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")
-text_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']}")
-                
-                # Display the image if available, otherwise show "Picture not available"
-                if plant_info["image_url"]:
+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:
                     st.image(plant_info["image_url"], caption=plant_info["name"], width=300)
-                else:
-                    st.write("**Picture not available**")
-
-if __name__ == "__main__":
-    main()
+                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.")