update chat app.py

app.py

@@ -1,55 +1,54 @@
 import streamlit as st
-# from IPython.display import display
-# from IPython.display import Markdown
+import google.generativeai as genai
 import os
 
 
 gemini_api_key = os.getenv("GEMINI_API_KEY")
 
-
-# def to_markdown(text):
-#   text = text.replace('•', '  *')
-#   return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True)
-
-import google.generativeai as genai
-
 genai.configure(api_key = gemini_api_key)
 
 model = genai.GenerativeModel('gemini-pro')
 
 
 
-prompt = st.chat_input("Say something")
-if prompt:
-    st.write(f"User has sent the following prompt: {prompt}")
-else:
-    prompt = "who are you?"
-response = model.generate_content(prompt)
-message = st.chat_message("ai")
-message.write(response.text)
+# prompt = st.chat_input("Say something")
+# if prompt:
+#     st.write(f"User has sent the following prompt: {prompt}")
+# else:
+#     prompt = "who are you?"
+# response = model.generate_content(prompt)
+# message = st.chat_message("ai")
+# message.write(response.text)
+
 
-# # Define a function that will be called when the user clicks the button
-# def greet_user():
-#     # Get the user's name from the text input
-#     name = st.text_input("Enter your name:")
+import string
+import random
 
-#     # Greet the user
-#     st.write(f"Hello, {name}!")
 
-# # Add a button to the app
-# st.button("Greet me!", greet_user)
+def randon_string() -> str:
+    return "".join(random.choices(string.ascii_uppercase + string.digits, k=10))
 
 
+def chat_actions():
+    st.session_state["chat_history"].append(
+        {"role": "user", "content": st.session_state["chat_input"]},
+    )
 
-# st.title("Hello! Streamlit App")
+    response = model.generate_content(st.session_state["chat_input"])
+    st.session_state["chat_history"].append(
+        {
+            "role": "assistant",
+            "content": response,
+        },  # This can be replaced with your chat response logic
+    )
 
-# x = st.slider('Select a value')
-# st.write(x, 'squared is', x * x)
 
+if "chat_history" not in st.session_state:
+    st.session_state["chat_history"] = []
 
-# st.title('AI Fitness Trainer: Squats Analysis')
 
+st.chat_input("Enter your message", on_submit=chat_actions, key="chat_input")
 
-# recorded_file = 'output_sample.mp4'
-# sample_vid = st.empty()
-# sample_vid.video(recorded_file)
+for i in st.session_state["chat_history"]:
+    with st.chat_message(name=i["role"]):
+        st.write(i["content"])