Update pages/python.py

pages/python.py

@@ -23,8 +23,8 @@ deep_seek = ChatHuggingFace(llm=deep_seek_skeleton,
                     max_new_tokens=150,
                     task = 'conversational')
 
-exp = ['<1', '1', '2', '3', '4', '5', '5+']
-selected = st.selectbox("Select experience:", exp)
+exp1 = ['<1', '1', '2', '3', '4', '5', '5+']
+exp = st.selectbox("Select experience:", exp1)
 if exp == '<1':
     experince = 'New bie mentor'
 elif exp == '1':
@@ -39,9 +39,42 @@ elif exp == '5':
     experince = '5'
 elif exp == '5+':
     experince = 'professional'
-user_input = st.text_input("Enter your query:")
 
 
 
-message = [SystemMessage(content='act as expert for creating a dashboard in streamlit using hugging face'),
-               HumanMessage(content=user_input)]
+
+selec = ['Python', 'Machine Learning', 'Deep Learning', 'Statistics', 'SQL', 'Excel']
+sub = st.selectbox("Select experience:", selec)
+if sub == '<1':
+    subject = 'New bie mentor'
+elif sub == '1':
+    subject = '1'
+elif sub == '2':
+    subject = '2'
+elif sub == '3':
+    subject = '3'
+elif sub == '4':
+    subject = '4'
+elif sub == '5':
+    subject = '5'
+elif sub == '5+':
+    subject = 'professional'
+
+
+
+    
+user_input = st.text_input("Enter your query:")
+l = []
+st.write(l)
+message = [SystemMessage(content=f'Act as {subject} mentor who has {experince} years of experience and the one who teaches in very friendly manner and also he explains everything within 150 words'),
+                   HumanMessage(content=user_input)]
+while user_input!='end':
+    l.append(user_input)
+    l.append(result.content)
+    st.write(l)
+    user_input = st.text_input("Enter your query:")
+    message = [SystemMessage(content=f'Act as {subject} mentor who has {experince} years of experience and the one who teaches in very friendly manner and also he explains everything within 150 words'),
+                   HumanMessage(content=user_input)]
+    result = deep_seek.invoke(message)
+    
+