Update app.py

app.py

@@ -14,12 +14,12 @@ llm = HuggingFaceHub(repo_id="suriya7/MaxMini-Instruct-248M",
                     })
 
 
-template = """
-Please Answer the Question: 
-previous chat: {previous_history}
-Human:{question}
-chatbot: 
-"""
+template = """You are a friendly chatbot called "MAXMINI" who give clear an well having a conversation with a human and you are created by suriya an AI Enthusiastic.
+previous chat: 
+{previous_history}
+Human:
+{question}
+Chatbot:"""
 
 prompt = PromptTemplate(template=template,input_variables=['question','previous_history'])
 
@@ -45,10 +45,10 @@ st.info("MaxMini-Instruct-248M is a T5 (Text-To-Text Transfer Transformer) model
 
 st.session_state['history'] = []
 
-if 'message' not in st.session_state:
-    st.session_state['message'] = ['Hey There! How Can I Assist You']
+if 'assistant' not in st.session_state:
+    st.session_state['assistant'] = ['Hey There! How Can I Assist You']
 
-st.session_state['past'] = [" "]
+st.session_state['human'] = [" "]
 
 
 # Create containers for chat history and user input
@@ -61,14 +61,14 @@ with container:
     if user_input:
         output = conversational_chat(user_input)
         # answer = response_generator(output)
-        st.session_state['past'].append(user_input)
-        st.session_state['message'].append(output)
+        st.session_state['human'].append(user_input)
+        st.session_state['assistant'].append(output)
         
         
-# Display chat history
-if st.session_state['message']:
+## Display chat history
+if st.session_state['assistant']:
     with response_container:
-        for i in range(len(st.session_state['message'])):
-            if i != 0:
-                message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="adventurer")
-            message(st.session_state["message"][i], key=str(i), avatar_style="bottts")
+        for i in range(len(st.session_state['assistant'])):
+            if i < len(st.session_state["human"]):  # Check if 'human' list has enough elements
+                message(st.session_state["human"][i], is_user=True, key=str(i) + '_user', avatar_style="adventurer")
+            message(st.session_state["assistant"][i], key=str(i), avatar_style="bottts")