Update app.py

app.py

@@ -158,9 +158,6 @@ def generate_rice_report(image_data, img_format, client):
         st.error(f"API communication error: {str(e)}")
         return None
 
-# ======================
-# UI COMPONENTS
-# ======================
 def display_main_interface():
     """Render primary application interface"""
     st.title("🌾 Rice Quality Analyzer")
@@ -174,28 +171,17 @@ def render_sidebar(client):
         uploaded_file = st.file_uploader("Select an image or video", type=ALLOWED_FILE_TYPES + ALLOWED_VIDEO_TYPES)
         
         if uploaded_file:
-            file_type = uploaded_file.type
-            if "video" in file_type:
-                frames = extract_video_frames(uploaded_file)
-                if frames:
-                    st.image(frames[0], caption="Extracted Frame for Analysis", use_column_width=True)
-                    frame_img = Image.fromarray(cv2.cvtColor(frames[0], cv2.COLOR_BGR2RGB))
-                    buffer = io.BytesIO()
-                    frame_img.save(buffer, format="JPEG")
-                    base64_image = base64.b64encode(buffer.getvalue()).decode('utf-8')
-                    report = generate_rice_report(base64_image, "jpeg", client)
-            else:
-                st.image(Image.open(uploaded_file), caption="Uploaded Image", use_column_width=True)
-                base64_image, img_format = process_image_data(uploaded_file)
-                report = generate_rice_report(base64_image, img_format, client)
+            base64_image, img_format = process_image_data(uploaded_file)
+            report = generate_rice_report(base64_image, img_format, client)
             
             if report:
-                st.markdown("### 📋 Analysis Report")
-                st.markdown(report)
+                st.session_state.analysis_result = report
+                st.rerun()
+    
+    if "analysis_result" in st.session_state:
+        st.markdown("### 📋 Analysis Report")
+        st.markdown(f'<div class="report-container"><div class="report-text">{st.session_state.analysis_result}</div></div>', unsafe_allow_html=True)
 
-# ======================
-# APPLICATION ENTRYPOINT
-# ======================
 def main():
     configure_application()
     groq_client = initialize_api_client()
@@ -203,4 +189,4 @@ def main():
     render_sidebar(groq_client)
 
 if __name__ == "__main__":
-    main()
+    main()