Update app.py

app.py

@@ -92,8 +92,6 @@ st.markdown("""
     </div>
 """, unsafe_allow_html=True)
 
-# Firebase Chat Functions
-
 def get_or_create_thread_id():
     doc_ref = db.collection("users").document(user_id)
     doc = doc_ref.get()
@@ -101,10 +99,7 @@ def get_or_create_thread_id():
         return doc.to_dict()["thread_id"]
     else:
         thread = client.beta.threads.create()
-        doc_ref.set({
-            "thread_id": thread.id,
-            "created_at": firestore.SERVER_TIMESTAMP
-        })
+        doc_ref.set({"thread_id": thread.id, "created_at": firestore.SERVER_TIMESTAMP})
         return thread.id
 
 def save_message(role, content):
@@ -124,10 +119,36 @@ def display_chat_history():
         else:
             st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>Carfind Assistant:</strong> {data['content']}</div>", unsafe_allow_html=True)
 
-# Tabs: AI Chat | Car Identifier
+def format_car_response(raw_text):
+    cleaned = re.sub(r'\*\*(.*?)\*\*', r'<strong>\1</strong>', raw_text)
+    cleaned = cleaned.replace("###", "<h4>").replace("\n", "<br>")
+    lines = raw_text.split("\n")
+    table_rows = []
+    intro = ""
+    for line in lines:
+        if "|" in line and not set(line.strip()) <= {"|", "-"}:
+            parts = [part.strip() for part in line.split("|") if part.strip()]
+            if len(parts) == 2:
+                label, value = parts
+                if value.lower() not in ["n/a", "null", "[specify mileage]", "[specify year]", "", "-"]:
+                    table_rows.append((label, value))
+        elif not intro and line.strip():
+            intro = line.strip()
+
+    html = f"<div style='margin-bottom: 10px; font-size: 12px; color: #003B6F;'>{intro}</div>"
+    if len(table_rows) >= 3:
+        html += "<table class='spec-table'><tr><th>Specification</th><th>Details</th></tr>"
+        for label, value in table_rows:
+            html += f"<tr><td>{label}</td><td>{value}</td></tr>"
+        html += "</table>"
+    else:
+        html += "<div style='color: grey;'>No detailed specifications available.</div>"
+    return html
+
+# Tabs
+
 tab1, tab2 = st.tabs(["AI Chat", "What car is that?"])
 
-# AI Chat Tab
 with tab1:
     input_col, clear_col = st.columns([9, 1])
     with input_col:
@@ -145,7 +166,6 @@ with tab1:
                 st.error(f"Failed to clear chat: {e}")
     thread_id = get_or_create_thread_id()
     display_chat_history()
-
     if user_input:
         client.beta.threads.messages.create(thread_id=thread_id, role="user", content=user_input)
         save_message("user", user_input)
@@ -163,7 +183,6 @@ with tab1:
         time.sleep(0.5)
         st.rerun()
 
-# Car Image Tab
 with tab2:
     uploaded_image = st.file_uploader("Upload an image of a car and let Ai identify it for you", type=["jpg", "jpeg", "png"])
     if uploaded_image:
@@ -180,7 +199,7 @@ with tab2:
                     role="user",
                     content=[
                         {"type": "image_file", "image_file": {"file_id": file_response.id}},
-                        {"type": "text", "text": "Please identify this car from the image and include a clean HTML-styled specification table without Year or Mileage fields, and avoid markdown or dotted lines."}
+                        {"type": "text", "text": "Please identify this car from the image and include a clean styled specifications as are available."}
                     ]
                 )
                 run = client.beta.threads.runs.create(thread_id=image_thread.id, assistant_id=assistant_id)
@@ -192,32 +211,8 @@ with tab2:
                         time.sleep(1)
                 messages = client.beta.threads.messages.list(thread_id=image_thread.id)
                 assistant_message = messages.data[0].content[0].text.value
-
-                # Clean base message
-                cleaned_message = re.sub(r'\*\*(.*?)\*\*', r'<strong>\1</strong>', assistant_message)
-                cleaned_message = cleaned_message.replace("###", "<h4 style='margin-bottom: 10px;'>").replace("\n", "<br>")
-
-                # Table builder (only if valid Markdown table found)
-                if '|' in cleaned_message and cleaned_message.count('|') > 2:
-                    lines = cleaned_message.split('<br>')
-                    html_table = "<table class='spec-table'>"
-                    has_table = False
-                    for line in lines:
-                        if '|' in line and '---' not in line and 'Year' not in line and 'Mileage' not in line:
-                            cells = line.strip('|').split('|')
-                            if len(cells) >= 2:
-                                has_table = True
-                                row_html = '<tr>' + ''.join(f'<td>{cell.strip()}</td>' for cell in cells) + '</tr>'
-                                html_table += row_html
-                        elif '|' not in line and has_table:
-                            html_table += "</table><p>" + line + "</p><table class='spec-table'>"
-                        elif '|' not in line:
-                            html_table += f"<p>{line}</p>"
-                    html_table += "</table>"
-                    cleaned_message = html_table if has_table else cleaned_message
-
+                formatted_html = format_car_response(assistant_message)
                 st.success("✅ Identification Complete")
-                st.markdown(f"<div class='car-spec-output'>{cleaned_message}</div>", unsafe_allow_html=True)
-
+                st.markdown(f"<div class='car-spec-output'>{formatted_html}</div>", unsafe_allow_html=True)
             except Exception as e:
                 st.error(f"❌ Error during image analysis: {str(e)}")