Update app.py

app.py

@@ -22,11 +22,11 @@ genai.configure(api_key=GOOGLE_API_KEY)
 model = genai.GenerativeModel('gemini-pro')
 
 
-def main():  # Wrap main logic in a function
-    if prompt := st.chat_input("Hi, how can I help you?"):
-        goal = prompt  # Capture the user's goal
-
-        goals_prompt = f"""Act as a personal assistant. Understand user intent from the point of view for asking a few questions. Ask to know more details about {goal}    :-  {goal}"""  
+def main():  
+    prompt = st.chat_input("Hi, how can I help you?") 
+    if prompt:        
+        goal = prompt   
+        goals_prompt = f"""Act as a personal assistant... {goal} """
         completion = model.generate_content(goals_prompt)  
 
         with st.chat_message("Assistant"):