create mainpy

main.py

@@ -0,0 +1,121 @@
+import streamlit as st
+from transformers import T5ForConditionalGeneration, T5Tokenizer
+from nltk.tokenize import sent_tokenize
+import nltk
+import json
+import os
+import time
+
+nltk.download('punkt')
+
+# Load Pre-Trained Model And Tokenizer
+tokenizer = T5Tokenizer.from_pretrained("t5-base")
+model = T5ForConditionalGeneration.from_pretrained("t5-base")
+
+def response_generator(msg_content):
+    lines = msg_content.split('\n')
+    for line in lines:
+        words = line.split()
+        for word in words:
+            yield word + " "
+            time.sleep(0.1)
+        yield "\n"
+
+def show_msgs():
+    for msg in st.session_state.messages:
+        role = msg["role"]
+        with st.chat_message(role):
+            st.write(msg["content"])
+
+def generate_response(text):
+    input_ids = tokenizer.encode(text, return_tensors="pt", max_length=512, truncation=True)
+    response_ids = model.generate(input_ids=input_ids, max_length=150, min_length=30, length_penalty=2.0, num_beams=4, early_stopping=True)
+    output = tokenizer.decode(response_ids[0], skip_special_tokens=True)
+    return output
+
+def format_messages_for_summary(messages):
+    return '\n'.join(f"{msg['role']}: {msg['content']}" for msg in messages)
+
+def save_chat():
+    if not os.path.exists('./Intermediate-Chats'):
+        os.makedirs('./Intermediate-Chats')
+    if st.session_state['messages']:
+        formatted_messages = format_messages_for_summary(st.session_state['messages'])
+        filename = f'./Intermediate-Chats/chat_{int(time.time())}.txt'
+        with open(filename, 'w') as f:
+            for message in st.session_state['messages']:
+                encoded_content = message['content'].replace('\n', '\\n')
+                f.write(f"{message['role']}: {encoded_content}\n")
+        st.session_state['messages'].clear()
+    else:
+        st.warning("No chat messages to save.")
+
+def load_saved_chats():
+    chat_dir = './Intermediate-Chats'
+    if os.path.exists(chat_dir):
+        files = os.listdir(chat_dir)
+        files.sort(key=lambda x: os.path.getmtime(os.path.join(chat_dir, x)), reverse=True)
+        for file_name in files:
+            display_name = file_name[:-4] if file_name.endswith('.txt') else file_name
+            if st.sidebar.button(display_name):
+                st.session_state['show_chats'] = False
+                st.session_state['is_loaded'] = True
+                load_chat(os.path.join(chat_dir, file_name))
+
+def load_chat(file_path):
+    st.session_state['messages'].clear()
+    with open(file_path, 'r') as file:
+        for line in file.readlines():
+            role, content = line.strip().split(': ', 1)
+            decoded_content = content.replace('\\n', '\n')
+            st.session_state['messages'].append({'role': role, 'content': decoded_content})
+
+def main():
+    st.title("T5 Chat Interface")
+
+    if 'messages' not in st.session_state:
+        st.session_state['messages'] = []
+    if 'show_chats' not in st.session_state:
+        st.session_state['show_chats'] = False
+
+    # File uploader
+    uploaded_files = st.file_uploader("Choose multiple files", type=["txt", "docx", "py", "java", "class", "php", "js", "css"], accept_multiple_files=True)
+    if uploaded_files:
+        for uploaded_file in uploaded_files:
+            file_name = uploaded_file.name
+            file_content = uploaded_file.read().decode("utf-8")
+            with st.expander(file_name):
+                st.write(file_content)
+
+    show_msgs()
+
+    user_input = st.chat_input("Enter your prompt:", key="1")
+    if user_input:
+        with st.chat_message("user"):
+            st.write(user_input)
+        st.session_state.messages.append({"role": "user", "content": user_input})
+        response = generate_response(user_input)
+        st.session_state.messages.append({"role": "assistant", "content": response})
+        with st.chat_message("assistant"):
+            st.write_stream(response_generator(response))
+
+    chatlog = format_messages_for_summary(st.session_state['messages'])
+    st.sidebar.download_button(
+        label="Download Chat Log",
+        data=chatlog,
+        file_name="chat_log.txt",
+        mime="text/plain"
+    )
+
+    if st.sidebar.button("Save Chat"):
+        save_chat()
+
+    if st.sidebar.button("New Chat"):
+        st.session_state['messages'].clear()
+
+    if st.sidebar.checkbox("Show/hide chat history", value=st.session_state['show_chats']):
+        st.sidebar.title("Previous Chats")
+        load_saved_chats()
+
+if __name__ == "__main__":
+    main()