Session States - app.py

app.py

@@ -3,12 +3,12 @@ from PIL import Image
 import torch
 import streamlit as st
 from streamlit_extras.stylable_container import stylable_container
-import backend as backend
+import backend
 from backend import *
 
 def chat_UI(model, tokenizer, device, uploaded_file):
     st.snow()
-    st.title('Welcome to :blue-background[***Icicle***!] :ice_cube: ')
+    st.header('Welcome to ***Icicle*** :ice_cube: ')
     st.divider()
 
     if uploaded_file is not None:
@@ -26,23 +26,28 @@ def process_input(model, tokenizer, device, uploaded_file, is_gif=False):
 
     user_input = st.session_state.get("user_input", "")
 
+    # Display the chat input bar
     user_input = st.chat_input("Say something")
 
     if user_input:
+        # Record user input
         chat_record("user", user_input)
 
+        # Generate answer
         if is_gif:
             response = process_gif(model, tokenizer, device, uploaded_file, user_input)
         else:
             response = generate_answer(model, tokenizer, device, uploaded_file, user_input)
 
+        # Record AI response
         chat_record("ai", response)
 
+        # Update conversation history
         conversation.append({'speaker': 'ai', 'message': response})
 
         st.session_state["conversation"] = conversation
 
-    # For display the AI responses in sequential order
+    # Display the AI responses in sequential order
     for i, chat in enumerate(conversation):
         if chat['speaker'] == 'ai':
             st.text_area(f"AI Response {i+1}", value=chat['message'],max_chars=None, key=None)
@@ -71,7 +76,7 @@ def local_css(file_name):
         st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
 
 def footer():
-    st.markdown('<div class="footer"><p>Developed with ❤ by <a style="display: block; text-align: center;" href="linkedin.com/in/taher-p-821817214" target="_blank">Taher!</a></p></div>',unsafe_allow_html=True)
+    st.markdown('<div class="footer"><p>Developed with ❤ by <a style="display: block; text-align: center;" href="linkedin.com/in/taher-p-821817214" target="_blank">Taher !</a></p></div>',unsafe_allow_html=True)
 
 
 def main():
@@ -82,4 +87,4 @@ def main():
     footer()
 
 if __name__=='__main__':
-    main()
+    main()