Update app.py

UI Enhancements (Already Included):
Custom CSS for a clean, modern look (light background, rounded inputs, styled buttons).
Improved typography with Roboto font and hierarchy (title, subtitle).
Responsive layout using st.container() and st.columns().
Chat-Like Output:
Session state (st.session_state.chat_history) stores input/output pairs as dictionaries.
Each entry is displayed with the user’s input and generated code in a conversational format.
Timestamps:
Added from datetime import datetime to include timestamps.
Each chat entry now includes a "time" key with the current time (e.g., "14:35:22"), shown next to "You" in the UI.
Clear History Button:
Added a "Clear History" button in the second column (col2), which resets st.session_state.chat_history to an empty list.
Displays a success message ("Chat history cleared!") for feedback.
Layout Adjustments:
Adjusted st.columns([1, 1]) to balance the "Generate" and "Clear History" buttons side by side.
Kept the chat history and tip below the input area for a natural flow.

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from datetime import datetime
 
 # Custom CSS for a Grok/ChatGPT-like look
 st.markdown("""
@@ -94,24 +95,31 @@ with st.container():
         height=150
     )
     
-    col1, col2 = st.columns([1, 5])
+    col1, col2 = st.columns([1, 1])  # Adjusted for two buttons
     with col1:
         if st.button("Generate"):
             if description.strip():
                 with st.spinner("Thinking..."):
                     generated_code = generate_code(description)
-                # Append to chat history
-                st.session_state.chat_history.append({"input": description, "output": generated_code})
+                # Append to chat history with timestamp
+                st.session_state.chat_history.append({
+                    "input": description,
+                    "output": generated_code,
+                    "time": datetime.now().strftime("%H:%M:%S")
+                })
             else:
                 st.warning("Please enter a description first!")
+    
     with col2:
-        st.empty()  # Spacer
+        if st.button("Clear History"):
+            st.session_state.chat_history = []
+            st.success("Chat history cleared!")
 
     # Display chat history
     if st.session_state.chat_history:
         st.write("### Chat History")
         for chat in st.session_state.chat_history:
-            st.markdown(f'<div class="chat-message"><strong>You:</strong> {chat["input"]}</div>', unsafe_allow_html=True)
+            st.markdown(f'<div class="chat-message"><strong>You ({chat["time"]}):</strong> {chat["input"]}</div>', unsafe_allow_html=True)
             st.markdown(f'<div class="code-output">{chat["output"]}</div>', unsafe_allow_html=True)
             st.markdown("---")  # Separator for readability