Update app.py

app.py

@@ -6,7 +6,6 @@ import os
 import uuid
 import firebase_admin
 from firebase_admin import credentials, firestore
-import re
 
 # 🔐 Firebase setup
 if not firebase_admin._apps:
@@ -47,38 +46,13 @@ st.markdown("""
     }
     .car-spec-output {
         font-family: "Segoe UI", sans-serif;
-        font-size: 10px;
+        font-size: 13px;
         background-color: #ffffff;
         color: #003B6F;
-        padding: 18px;
+        padding: 20px;
         border-radius: 10px;
-        box-shadow: 0 4px 12px rgba(0,0,0,0.05);
-        border-left: 4px solid #0071BC;
-        line-height: 1.5;
-    }
-    .spec-table {
-        width: 100%;
-        border-collapse: collapse;
-        margin-top: 12px;
-        margin-bottom: 16px;
-        font-size: 10px;
-    }
-    .spec-table th, .spec-table td {
-        border: 1px solid #ddd;
-        padding: 6px 8px;
-        text-align: left;
-    }
-    .spec-table th {
-        background-color: #0071BC;
-        color: white;
-    }
-    .spec-table td {
-        background-color: #f7f9fc;
-    }
-    .icon {
-        width: 14px;
-        vertical-align: middle;
-        margin-right: 5px;
+        border-left: 5px solid #0071BC;
+        line-height: 1.6;
     }
     </style>
 """, unsafe_allow_html=True)
@@ -119,32 +93,6 @@ 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)
 
-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?"])
@@ -199,7 +147,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 styled specifications as are available."}
+                        {"type": "text", "text": "Identify this car image and respond using the markdown output format shown in your instructions — include all sections (🟢 Identified Vehicle, 📋 Details, 🚘 Overview, 🔎 Recommended For) without follow-ups or links."}
                     ]
                 )
                 run = client.beta.threads.runs.create(thread_id=image_thread.id, assistant_id=assistant_id)
@@ -211,8 +159,7 @@ 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
-                formatted_html = format_car_response(assistant_message)
                 st.success("✅ Identification Complete")
-                st.markdown(f"<div class='car-spec-output'>{formatted_html}</div>", unsafe_allow_html=True)
+                st.markdown(f"<div class='car-spec-output'>{assistant_message}</div>", unsafe_allow_html=True)
             except Exception as e:
                 st.error(f"❌ Error during image analysis: {str(e)}")