Update app.py

app.py

@@ -3,17 +3,28 @@ import streamlit as st
 import os 
 
 # Set your API key
-palm.configure(api_key=os.environ['PALM_KEY'])
+palm.configure(api_key = os.environ['PALM_KEY'])
 
 # Select the PaLM 2 model
 model = 'models/text-bison-001'
 
 # Generate text
-prompt = "Write a poem about a cat."
-completion = palm.generate_text(model=model, prompt=prompt, temperature=0.5, max_output_tokens=800)
+if prompt := st.chat_input("Hi, I can help you manage your daily tasks."):
+    enprom = f"""Understand whether user is asking to create a task or trying to have a general conversation 
+    or is saying something which relates to a task creation thing and can be further discussed to know about task details.
+    If user is asking to create task then take all details for creating a task and send as a table for 4 columns i.e Task title, time, repetation, status.
+    Else if user is trying to have just a normal general conversation, then give a reply accordingly.
+    or if user is talking about something that can be related to a task and ask more question from the user to get more clarity about task details and show me the the discussions and question you had and also give me a table with those 4 columns for task details.
+    """
+    completion = palm.generate_text(model=model, prompt=prompt, temperature=0.5, max_output_tokens=800)
+
+# response = palm.chat(messages=["Hello."])
+# print(response.last) #  'Hello! What can I help you with?'
+# response.reply("Can you tell me a joke?")
 
 # Print the generated text
-st.write(completion.result)
+    with st.chat_message("Assostant"):
+        st.write(completion.result)