Update app.py

app.py

@@ -9,11 +9,25 @@ from keras.applications.inception_v3 import preprocess_input
 import os
 from PIL import Image
 import io
+from datetime import datetime
+import pandas as pd
 st.set_page_config(page_title="Wall Defect Classifier", layout="centered")
-
+if "dpoc_list" not in st.session_state:
+    st.session_state.dpoc_list = []
 # Set OpenAI API Key
 openai.api_key = os.getenv('OPEN_AI')
-
+def add_dpoc_entry(title, description):
+    now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    entry = {
+        "id": len(st.session_state.dpoc_list) + 1,
+        "dpoc_category_id": 1,  # You can dynamically set this if needed
+        "title": title,
+        "description": description,
+        "created_at": now,
+        "updated_at": now,
+        "deleted_at": ""
+    }
+    st.session_state.dpoc_list.append(entry)
 # Defect categories
 class_labels = [
     "Floor Cracks",
@@ -95,23 +109,34 @@ if category_choice == "Flooring":
                 f"Description: <A concise, technical description in 100 words or less>"
             )
         
-            st.subheader("Generating AI-Based Analysis...")
-            try:
-                response = openai.ChatCompletion.create(
-                    model="gpt-4o",
-                    messages=[
-                        {
-                            "role": "user",
-                            "content": [
-                                {"type": "text", "text": ai_prompt},
-                                {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{compressed_base64}"}}
-                            ]
-                        }
-                    ],
-                    max_tokens=300,
-                )
-                ai_description = response.choices[0].message.content
-                st.subheader("📝 AI-Generated Defect Description")
-                st.text_area("Output", value=ai_description.strip(), height=250)
-            except Exception as e:
-              st.error(f"❌ An error occurred while generating the description:\n{e}")
+            with st.spinner("📝 Generating AI description..."):
+                try:
+                    response = openai.ChatCompletion.create(
+                        model="gpt-4o",
+                        messages=[
+                            {
+                                "role": "user",
+                                "content": [
+                                    {"type": "text", "text": ai_prompt},
+                                    {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{compressed_base64}"}}
+                                ]
+                            }
+                        ],
+                        max_tokens=300,
+                    )
+                    ai_description = response.choices[0].message.content
+                    st.subheader("📝 AI-Generated Defect Description")
+                    st.text_area("Output", value=ai_description.strip(), height=250)
+                except Exception as e:
+                  st.error(f"❌ An error occurred while generating the description:\n{e}")
+            add_dpoc_entry(class_name, ai_description.strip())
+            st.subheader("📋 DPOC Log")
+            if st.session_state.dpoc_list:
+                dpoc_df = pd.DataFrame(st.session_state.dpoc_list)
+                st.dataframe(dpoc_df, use_container_width=True)
+                st.download_button("📥 Download CSV", dpoc_df.to_csv(index=False), "dpoc_log.csv", "text/csv")
+            else:
+                st.info("No entries yet. Upload an image to start.")
+        
+
+