Update app.py

app.py

@@ -192,21 +192,32 @@ 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>")
-                if '|' in cleaned_message:
+
+                # 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('|')
-                            row_html = '<tr>' + ''.join(f'<td>{cell.strip()}</td>' for cell in cells) + '</tr>'
-                            html_table += row_html
+                            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"</table><p>{line}</p><table class='spec-table'>"
+                            html_table += f"<p>{line}</p>"
                     html_table += "</table>"
-                    cleaned_message = html_table
+                    cleaned_message = html_table if has_table else cleaned_message
+
                 st.success("✅ Identification Complete")
                 st.markdown(f"<div class='car-spec-output'>{cleaned_message}</div>", unsafe_allow_html=True)
+
             except Exception as e:
-                st.error(f"❌ Error during image analysis: {str(e)}")
+                st.error(f"❌ Error during image analysis: {str(e)}")