Spaces:
Runtime error
Runtime error
Commit
Β·
97e1148
1
Parent(s):
deb2e3c
Upload 2 files
Browse files- coding.py +127 -0
- requirements.txt +7 -0
coding.py
ADDED
@@ -0,0 +1,127 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
import os
|
3 |
+
import pickle
|
4 |
+
import time
|
5 |
+
import g4f
|
6 |
+
|
7 |
+
st.set_page_config(page_title="CODING ASSISTANT")
|
8 |
+
|
9 |
+
st.markdown(
|
10 |
+
"""
|
11 |
+
<style>
|
12 |
+
.title {
|
13 |
+
text-align: center;
|
14 |
+
font-size: 2em;
|
15 |
+
font-weight: bold;
|
16 |
+
}
|
17 |
+
</style>
|
18 |
+
<div class="title"> π¨βπ» CODING ASSISTANT β</div>
|
19 |
+
""",
|
20 |
+
unsafe_allow_html=True
|
21 |
+
)
|
22 |
+
# Load and Save Conversations
|
23 |
+
conversations_file = "conversations.pkl"
|
24 |
+
|
25 |
+
|
26 |
+
@st.cache_data
|
27 |
+
def load_conversations():
|
28 |
+
try:
|
29 |
+
with open(conversations_file, "rb") as f:
|
30 |
+
return pickle.load(f)
|
31 |
+
except (FileNotFoundError, EOFError):
|
32 |
+
return []
|
33 |
+
|
34 |
+
|
35 |
+
def save_conversations(conversations):
|
36 |
+
temp_conversations_file = conversations_file
|
37 |
+
with open(temp_conversations_file, "wb") as f:
|
38 |
+
pickle.dump(conversations, f)
|
39 |
+
os.replace(temp_conversations_file, conversations_file)
|
40 |
+
|
41 |
+
|
42 |
+
if 'conversations' not in st.session_state:
|
43 |
+
st.session_state.conversations = load_conversations()
|
44 |
+
|
45 |
+
if 'current_conversation' not in st.session_state:
|
46 |
+
st.session_state.current_conversation = [{"role": "assistant", "content": "How may I assist you today?"}]
|
47 |
+
|
48 |
+
|
49 |
+
def truncate_string(s, length=30):
|
50 |
+
return s[:length].rstrip() + "..." if len(s) > length else s
|
51 |
+
|
52 |
+
|
53 |
+
def display_chats_sidebar():
|
54 |
+
with st.sidebar.container():
|
55 |
+
st.header('Settings')
|
56 |
+
col1, col2 = st.columns([1, 1])
|
57 |
+
|
58 |
+
with col1:
|
59 |
+
if col1.button('Start New Chat', key="new_chat"):
|
60 |
+
st.session_state.current_conversation = []
|
61 |
+
st.session_state.conversations.append(st.session_state.current_conversation)
|
62 |
+
|
63 |
+
with col2:
|
64 |
+
if col2.button('Clear All Chats', key="clear_all"):
|
65 |
+
st.session_state.conversations = []
|
66 |
+
st.session_state.current_conversation = []
|
67 |
+
|
68 |
+
|
69 |
+
with st.sidebar.container():
|
70 |
+
st.header('Conversations')
|
71 |
+
for idx, conversation in enumerate(st.session_state.conversations):
|
72 |
+
if conversation:
|
73 |
+
chat_title_raw = next((msg["content"] for msg in conversation if msg["role"] == "user"), "New Chat")
|
74 |
+
chat_title = truncate_string(chat_title_raw)
|
75 |
+
if st.sidebar.button(f"{chat_title}", key=f"chat_button_{idx}"):
|
76 |
+
st.session_state.current_conversation = st.session_state.conversations[idx]
|
77 |
+
|
78 |
+
|
79 |
+
|
80 |
+
def main_app():
|
81 |
+
for message in st.session_state.current_conversation:
|
82 |
+
with st.chat_message(message["role"]):
|
83 |
+
st.write(message["content"])
|
84 |
+
|
85 |
+
def generate_response(prompt_input):
|
86 |
+
string_dialogue = '''
|
87 |
+
You are a virtual coding assistant, specialized in providing help with programming languages, software development concepts,
|
88 |
+
debugging tips, and code optimization strategies. Your expertise covers a wide range of programming languages including but not
|
89 |
+
limited to Python, JavaScript, Java, C++, and HTML/CSS. You can assist with understanding programming concepts, syntax, algorithms,
|
90 |
+
and best practices. You also offer guidance on troubleshooting and debugging code, as well as improving code efficiency and readability.
|
91 |
+
|
92 |
+
Please note that your role is to assist with coding-related inquiries only. You do not engage in non-programming related discussions or provide personal opinions on matters outside of software development.
|
93 |
+
|
94 |
+
Human:
|
95 |
+
'''
|
96 |
+
for dict_message in st.session_state.current_conversation:
|
97 |
+
string_dialogue += dict_message["role"].capitalize() + ": " + dict_message["content"] + "\\n\\n"
|
98 |
+
|
99 |
+
prompt = f"{string_dialogue}\n {prompt_input} Assistant: "
|
100 |
+
response_generator = g4f.ChatCompletion.create(
|
101 |
+
model="gpt-3.5-turbo",
|
102 |
+
messages=[{"role": "user", "content": prompt}],
|
103 |
+
stream=True,
|
104 |
+
)
|
105 |
+
return response_generator
|
106 |
+
|
107 |
+
if prompt := st.chat_input('Send a Message'):
|
108 |
+
st.session_state.current_conversation.append({"role": "user", "content": prompt})
|
109 |
+
with st.chat_message("user"):
|
110 |
+
st.write(prompt)
|
111 |
+
|
112 |
+
with st.chat_message("assistant"):
|
113 |
+
with st.spinner("Thinking..."):
|
114 |
+
response = generate_response(prompt)
|
115 |
+
placeholder = st.empty()
|
116 |
+
full_response = ''
|
117 |
+
for item in response:
|
118 |
+
full_response += item
|
119 |
+
time.sleep(0.003)
|
120 |
+
placeholder.markdown(full_response)
|
121 |
+
placeholder.markdown(full_response)
|
122 |
+
st.session_state.current_conversation.append({"role": "assistant", "content": full_response})
|
123 |
+
save_conversations(st.session_state.conversations)
|
124 |
+
|
125 |
+
|
126 |
+
display_chats_sidebar()
|
127 |
+
main_app()
|
requirements.txt
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
gpt4all==1.0.8
|
2 |
+
streamlit
|
3 |
+
g4f
|
4 |
+
docx2txt
|
5 |
+
pypdf2
|
6 |
+
pytesseract
|
7 |
+
pdf2image
|