Update app.py

app.py

@@ -25,6 +25,13 @@ client = Groq(
     api_key=os.environ['GROQ_API_KEY'],
 )
 
+# Read saved prompts from file
+with open("saved_prompts.txt", "r") as f:
+    saved_prompts = f.read().split("<|>")
+
+prompt_names = [p.split(" ", 1)[0] for p in saved_prompts]
+prompt_map = {name: prompt for name, prompt in zip(prompt_names, saved_prompts)}
+
 # Initialize chat history and selected model
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -55,6 +62,13 @@ with col1:
         format_func=lambda x: models[x]["name"],
         index=0,  # Default to the first model in the list
     )
+# Add prompt dropdown
+prompt_option = st.selectbox("Choose a prompt:", options=prompt_names)
+
+if not prompt_option:
+    prompt = ""
+else:
+    prompt = prompt_map[prompt_option]
 
 # Detect model change and clear chat history if model has changed
 if st.session_state.selected_model != model_option:
@@ -81,36 +95,18 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
-# Function to save prompt to a file
-def save_prompt(prompt):
-    with open("saved_prompts.txt", "a") as file:
-        file.write(prompt + "\n")
-
-# Function to delete prompt from file
-def delete_prompt(prompt_index):
-    with open("saved_prompts.txt", "r") as file:
-        prompts = file.readlines()
-    with open("saved_prompts.txt", "w") as file:
-        for i, p in enumerate(prompts):
-            if i != prompt_index:
-                file.write(p)
-
-
 def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
     """Yield chat response content from the Groq API response."""
     for chunk in chat_completion:
         if chunk.choices[0].delta.content:
             yield chunk.choices[0].delta.content
 
-if prompt := st.chat_input("Enter your prompt here..."):
+if prompt := st.chat_input("Enter your prompt here...", value=prompt):
     st.session_state.messages.append({"role": "user", "content": prompt})
 
     with st.chat_message("user", avatar="❓"):  
         st.markdown(prompt)
 
-    if st.button("Save"):
-        save_prompt(prompt)
-
     # Fetch response from Groq API
     try:
         chat_completion = client.chat.completions.create(
@@ -140,15 +136,4 @@ if prompt := st.chat_input("Enter your prompt here..."):
         combined_response = "\n".join(str(item) for item in full_response)
         st.session_state.messages.append(
             {"role": "assistant", "content": combined_response}
-        )
-
-# Collapsible section for prompt management
-with st.expander("Prompt Management", expanded=False):
-    if os.path.exists("saved_prompts.txt"):
-        with open("saved_prompts.txt", "r") as file:
-            saved_prompts = file.readlines()
-        for i, prompt in enumerate(saved_prompts):
-            delete_button = st.button("Delete", key=f"delete_{i}")
-            st.write(f"{i + 1}. {prompt.strip()}")
-            if delete_button:
-                delete_prompt(i)
+        )