Update app.py

app.py

@@ -35,25 +35,6 @@ st.markdown("""
     .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; margin-bottom: 10px; }
     .stChatMessage[data-testid="stChatMessage-user"] { background: #f0f0f0; color: #000000; }
     .stChatMessage[data-testid="stChatMessage-assistant"] { background: #D6E9FE; color: #000000; }
-    @keyframes bounceIn {
-        0% { transform: scale(0.7); opacity: 0; }
-        60% { transform: scale(1.1); opacity: 1; }
-        80% { transform: scale(0.95); }
-        100% { transform: scale(1); }
-    }
-    .carfind-logo {
-        animation: bounceIn 0.6s ease-out;
-    }
-    .car-spec-output {
-        font-family: "Segoe UI", sans-serif;
-        font-size: 13px;
-        background-color: #ffffff;
-        color: #003B6F;
-        padding: 20px;
-        border-radius: 10px;
-        border-left: 5px solid #0071BC;
-        line-height: 1.6;
-    }
     </style>
 """, unsafe_allow_html=True)
 
@@ -96,78 +77,38 @@ 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 for Chat and Car Identification
-tab1, tab2 = st.tabs(["AI Chat", "What car is that?"])
+# 🚀 Main Assistant Chat Interface
+input_col, clear_col = st.columns([9, 1])
+with input_col:
+    user_input = st.chat_input("Type your message here...")
+with clear_col:
+    if st.button("🗑️", key="clear-chat", help="Clear Chat"):
+        try:
+            user_doc_ref = db.collection("users").document(user_id)
+            for msg in user_doc_ref.collection("messages").stream():
+                msg.reference.delete()
+            user_doc_ref.delete()
+            st.session_state.clear()
+            st.rerun()
+        except Exception as e:
+            st.error(f"Failed to clear chat: {e}")
 
-# Tab 1: AI Chat
-with tab1:
-    input_col, clear_col = st.columns([9, 1])
-    with input_col:
-        user_input = st.chat_input("Type your message here...")
-    with clear_col:
-        if st.button("🗑️", key="clear-chat", help="Clear Chat"):
-            try:
-                user_doc_ref = db.collection("users").document(user_id)
-                for msg in user_doc_ref.collection("messages").stream():
-                    msg.reference.delete()
-                user_doc_ref.delete()
-                st.session_state.clear()
-                st.rerun()
-            except Exception as e:
-                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)
-        with st.spinner("Thinking and typing... 💭"):
-            run = client.beta.threads.runs.create(thread_id=thread_id, assistant_id=assistant_id)
-            while True:
-                run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
-                if run_status.status == "completed":
-                    break
-                time.sleep(1)
-        messages_response = client.beta.threads.messages.list(thread_id=thread_id)
-        latest_response = sorted(messages_response.data, key=lambda x: x.created_at)[-1]
-        assistant_message = latest_response.content[0].text.value
-        save_message("assistant", assistant_message)
-        time.sleep(0.5)
-        st.rerun()
+thread_id = get_or_create_thread_id()
+display_chat_history()
 
-# Tab 2: "What Car is That?"
-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:
-        col1, col2 = st.columns([1.2, 1.8])
-        with col1:
-            image = Image.open(uploaded_image)
-            st.image(image, caption="Uploaded Image", use_container_width=True)
-        with col2:
-            try:
-                # Create a new thread for image processing
-                image_thread = client.beta.threads.create()
-                file_response = client.files.create(file=uploaded_image, purpose="assistants")
-                client.beta.threads.messages.create(
-                    thread_id=image_thread.id,
-                    role="user",
-                    content=[{
-                        "type": "image_file", 
-                        "image_file": {"file_id": file_response.id}
-                    }, {
-                        "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)
-                with st.spinner("🔍 Analyzing image and identifying the car..."):
-                    while True:
-                        run_status = client.beta.threads.runs.retrieve(thread_id=image_thread.id, run_id=run.id)
-                        if run_status.status == "completed":
-                            break
-                        time.sleep(1)
-                messages = client.beta.threads.messages.list(thread_id=image_thread.id)
-                assistant_message = messages.data[0].content[0].text.value
-                st.success("✅ Identification Complete")
-                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)}")
+if user_input:
+    client.beta.threads.messages.create(thread_id=thread_id, role="user", content=user_input)
+    save_message("user", user_input)
+    with st.spinner("Thinking and typing... 💭"):
+        run = client.beta.threads.runs.create(thread_id=thread_id, assistant_id=assistant_id)
+        while True:
+            run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
+            if run_status.status == "completed":
+                break
+            time.sleep(1)
+    messages_response = client.beta.threads.messages.list(thread_id=thread_id)
+    latest_response = sorted(messages_response.data, key=lambda x: x.created_at)[-1]
+    assistant_message = latest_response.content[0].text.value
+    save_message("assistant", assistant_message)
+    time.sleep(0.5)
+    st.rerun()