Update app.py

app.py

@@ -12,12 +12,12 @@ from reportlab.lib.styles import getSampleStyleSheet
 # ======================
 # CONFIGURATION SETTINGS
 # ======================
-st.set_page_config(
-    page_title="Rice Quality Analyzer",
-    page_icon="🌾",
-    layout="wide",
-    initial_sidebar_state="expanded"
-)
+PAGE_CONFIG = {
+    "page_title": "Rice Quality Analyzer",
+    "page_icon": "🌾",
+    "layout": "wide",
+    "initial_sidebar_state": "expanded"
+}
 
 ALLOWED_FILE_TYPES = ['png', 'jpg', 'jpeg']
 
@@ -26,11 +26,11 @@ CSS_STYLES = """
     .main { background-color: #f4f9f9; color: #000000; }
     .sidebar .sidebar-content { background-color: #d1e7dd; }
     .stTextInput textarea { color: #000000 !important; }
-    .stButton>button {
-        background-color: #21eeef;
-        color: white;
-        font-size: 16px;
-        border-radius: 5px;
+    .stButton>button { 
+        background-color: #21eeef; 
+        color: white; 
+        font-size: 16px; 
+        border-radius: 5px; 
     }
     .report-container {
         background-color: #ffffff;
@@ -48,22 +48,28 @@ CSS_STYLES = """
     }
 </style>
 """
-st.markdown(CSS_STYLES, unsafe_allow_html=True)
 
 # ======================
 # CORE FUNCTIONS
 # ======================
+def configure_application():
+    """Initialize application settings and styling"""
+    st.set_page_config(**PAGE_CONFIG)
+    st.markdown(CSS_STYLES, unsafe_allow_html=True)
+
 def initialize_api_client():
-    """Initialize and validate the Groq API client."""
+    """Create and validate Groq API client"""
     load_dotenv()
     api_key = os.getenv("GROQ_API_KEY")
+    
     if not api_key:
-        st.error("API key not found. Please check .env configuration.")
+        st.error("API key not found. Please verify .env configuration.")
         st.stop()
+    
     return Groq(api_key=api_key)
 
 def process_image_data(uploaded_file):
-    """Convert image to base64 encoded string."""
+    """Convert image to base64 encoded string"""
     try:
         image = Image.open(uploaded_file)
         buffer = io.BytesIO()
@@ -74,49 +80,46 @@ def process_image_data(uploaded_file):
         return None, None
 
 def generate_pdf_report(report_text):
-    """Generate a PDF report from the analysis text."""
+    """Generate PDF document from report text"""
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(buffer, pagesize=letter)
     styles = getSampleStyleSheet()
-    story = [
-        Paragraph("<b>Rice Quality Report</b>", styles['Title']),
-        Spacer(1, 12),
-        Paragraph(report_text.replace('\n', '<br/>'), styles['BodyText'])
-    ]
+    story = []
+    
+    title = Paragraph("<b>Rice Quality Report</b>", styles['Title'])
+    story.append(title)
+    story.append(Spacer(1, 12))
+    
+    content = Paragraph(report_text.replace('\n', '<br/>'), styles['BodyText'])
+    story.append(content)
+    
     doc.build(story)
     buffer.seek(0)
     return buffer
 
 def generate_rice_report(uploaded_file, client):
-    """Generate AI-powered rice quality analysis."""
+    """Generate AI-powered rice quality analysis"""
     base64_image, img_format = process_image_data(uploaded_file)
+    
     if not base64_image:
         return None
-    
+
     image_url = f"data:image/{img_format.lower()};base64,{base64_image}"
-    
+
     try:
         response = client.chat.completions.create(
             model="llama-3.2-11b-vision-preview",
-            messages=[
-                {"role": "user", "content": [
+            messages=[{
+                "role": "user",
+                "content": [
                     {"type": "text", "text": (
                         "Analyze the rice grain image and provide a detailed report including:\n"
                         "1. Rice type classification\n2. Quality assessment (broken grains %, discoloration %, impurities %)\n"
-                        "3. Foreign object detection\n4. Size and shape consistency\n5. Recommendations for processing or improvement\n"
-                        "6. Additional metrics:\n"
-                        "   - % Broken kernel (Accuracy < 0.5%)\n"
-                        "   - Size analysis per each kernel (Accuracy < 50 micron):\n"
-                        "     - Length, Width, Area, Length to Width Ratio\n"
-                        "   - Abnormal color and damage kernel recognition:\n"
-                        "     - Chalky, Black spots, Yellow (Heat damage), Red (Heat damage), Green kernels\n"
-                        "   - Kernel count\n"
-                        "   - 1,000 kernels weight\n"
-                        "   - Group sizing information"
+                        "3. Foreign object detection\n4. Size and shape consistency\n5. Recommendations for processing or improvement"
                     )},
                     {"type": "image_url", "image_url": {"url": image_url}},
-                ]}
-            ],
+                ]
+            }],
             temperature=0.2,
             max_tokens=400,
             top_p=0.5
@@ -130,23 +133,16 @@ def generate_rice_report(uploaded_file, client):
 # UI COMPONENTS
 # ======================
 def display_main_interface():
-    """Render primary application interface."""
+    """Render primary application interface"""
     st.title("🌾 Rice Quality Analyzer")
     st.subheader("AI-Powered Rice Grain Inspection")
     st.markdown("---")
 
-    st.markdown("### Key Benefits:")
-    st.markdown("- Reduces inspection time by a factor of 10")
-    st.markdown("- Simple 3-step process: Load sample → Click start → Get results")
-    st.markdown("- Results stored in a database for traceability and further analysis")
-    st.markdown("- Integration with 3rd party instruments for higher-level data analysis")
-    st.markdown("- No minimum sample size required")
-    st.markdown("- No complicated sample preparation needed")
-    
-    if 'analysis_result' in st.session_state:
+    # Display analysis results
+    if st.session_state.get('analysis_result'):
         st.markdown("### 📋 Analysis Report")
         st.markdown(
-            f'<div class="report-container"><div class="report-text">{st.session_state.analysis_result}</div></div>',
+            f'<div class="report-container"><div class="report-text">{st.session_state.analysis_result}</div></div>', 
             unsafe_allow_html=True
         )
         pdf_report = generate_pdf_report(st.session_state.analysis_result)
@@ -156,15 +152,48 @@ def display_main_interface():
             file_name="rice_quality_report.pdf",
             mime="application/pdf"
         )
-    
+
     if st.button("Clear Analysis 🗑️"):
         st.session_state.pop('analysis_result', None)
         st.rerun()
 
+def render_sidebar(client):
+    """Create sidebar interface elements"""
+    with st.sidebar:
+        st.markdown("### Features")
+        st.markdown("""
+        - **Rice Type Classification** (e.g., Basmati, Jasmine, Indica)
+        - **Quality Check** (Broken grains %, impurities %, discoloration %)
+        - **Foreign Object Detection** (Husks, stones, debris)
+        - **Grain Size & Shape Analysis**
+        - **Processing Recommendations**
+        """)
+        st.markdown("---")
+        
+        st.subheader("Upload Rice Image")
+        uploaded_file = st.file_uploader(
+            "Select an image of rice grains", 
+            type=ALLOWED_FILE_TYPES
+        )
+        
+        if uploaded_file:
+            st.image(Image.open(uploaded_file), caption="Uploaded Image", use_column_width=True)
+            if st.button("Analyze Rice Quality 🔍"):
+                with st.spinner("Processing image... This may take a few seconds."):
+                    report = generate_rice_report(uploaded_file, client)
+                    st.session_state.analysis_result = report
+                    st.rerun()
+
+# ======================
+# APPLICATION ENTRYPOINT
+# ======================
 def main():
-    """Primary application controller."""
+    """Primary application controller"""
+    configure_application()
     groq_client = initialize_api_client()
+    
     display_main_interface()
+    render_sidebar(groq_client)
 
 if __name__ == "__main__":
-    main()
+    main()