Upload 2 files

leaf-disease-huggingface.py

@@ -0,0 +1,144 @@
+import streamlit as st
+import requests
+import base64
+import os
+from PIL import Image
+import io
+
+# Set page configuration
+st.set_page_config(
+    page_title="Leaf Disease Identifier",
+    page_icon="🍃",
+    layout="centered"
+)
+
+# Custom CSS for styling
+st.markdown("""
+<style>
+.main-title {
+    font-size: 36px;
+    color: #2C5F2D;
+    text-align: center;
+    margin-bottom: 20px;
+}
+.subtitle {
+    font-size: 18px;
+    color: #4A6741;
+    text-align: center;
+    margin-bottom: 30px;
+}
+.stButton>button {
+    background-color: #2C5F2D;
+    color: white;
+    width: 100%;
+    border: none;
+    padding: 10px;
+    border-radius: 5px;
+}
+.stButton>button:hover {
+    background-color: #4A6741;
+}
+</style>
+""", unsafe_allow_html=True)
+
+# Title and Description
+st.markdown('<h1 class="main-title">🍃 Leaf Disease Identifier</h1>', unsafe_allow_html=True)
+st.markdown('<p class="subtitle">Upload a leaf image and get instant disease analysis</p>', unsafe_allow_html=True)
+
+def encode_image(image):
+    """Encode image to base64"""
+    buffered = io.BytesIO()
+    image.save(buffered, format="PNG")
+    return base64.b64encode(buffered.getvalue()).decode('utf-8')
+
+def analyze_leaf_disease(image):
+    """Analyze leaf image using Claude API"""
+    # Hugging Face will require you to add API key as an environment variable
+    api_key = os.getenv("ANTHROPIC_API_KEY")
+    
+    if not api_key:
+        st.error("API Key not configured. Please set up Anthropic API key.")
+        return None
+
+    headers = {
+        "Content-Type": "application/json",
+        "anthropic-version": "2023-06-01",
+        "X-API-Key": api_key
+    }
+
+    payload = {
+        "model": "claude-3-5-sonnet-20240620",
+        "max_tokens": 1000,
+        "messages": [
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "image",
+                        "source": {
+                            "type": "base64",
+                            "media_type": "image/png",
+                            "data": encode_image(image)
+                        }
+                    },
+                    {
+                        "type": "text",
+                        "text": """Analyze this leaf image in detail. 
+                        Identify:
+                        1. Plant species (if possible)
+                        2. Specific disease or health condition
+                        3. Detailed symptoms
+                        4. Potential causes
+                        5. Recommended treatment or management strategies
+                        
+                        Provide a comprehensive and clear explanation."""
+                    }
+                ]
+            }
+        ]
+    }
+
+    try:
+        response = requests.post(
+            "https://api.anthropic.com/v1/messages", 
+            headers=headers, 
+            json=payload
+        )
+        response.raise_for_status()
+        result = response.json()
+        return result['content'][0]['text']
+    
+    except requests.exceptions.RequestException as e:
+        st.error(f"API Request Error: {e}")
+        return None
+    except KeyError:
+        st.error("Unexpected API response format")
+        return None
+
+def main():
+    # Image upload
+    uploaded_file = st.file_uploader(
+        "Upload a leaf image", 
+        type=["jpg", "jpeg", "png"],
+        help="Upload a clear image of a leaf for disease analysis"
+    )
+
+    if uploaded_file is not None:
+        # Display uploaded image
+        image = Image.open(uploaded_file)
+        st.image(image, caption='Uploaded Leaf Image', use_column_width=True)
+
+        # Analysis button
+        if st.button('Analyze Leaf Disease'):
+            with st.spinner('Analyzing image... This might take a moment'):
+                analysis = analyze_leaf_disease(image)
+                
+                if analysis:
+                    st.success('Analysis Complete!')
+                    st.markdown("### 🔬 Leaf Disease Analysis")
+                    st.write(analysis)
+                else:
+                    st.error("Failed to analyze the image. Please try again.")
+
+if __name__ == "__main__":
+    main()

requirements-huggingface.txt

@@ -0,0 +1,3 @@
+streamlit==1.33.0
+requests==2.31.0
+Pillow==10.3.0