Update chat.py

chat.py

@@ -1,26 +1,47 @@
-# from transformers import pipeline, Conversation
-# # import streamlit_option_menu
-# import streamlit as st
-
-# def Chat():
-
-#     query = st.chat_input("Enter your query")
-#     convo = pipeline("conversational")
-#     oracle = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
-#     usrinput = Conversation(query)
-#     chitchat = convo(usrinput)
-#     ans = oracle(
-#         query,
-#         candidate_labels=["logout"])
-
-#     if ans["scores"][0] > 0.85:
-#         st.session_state["user"] = "visitor"
-#         with st.chat_message("assistant"):
-#             "You are now living in dream" 
-#         st.experimental_rerun()
-#     else:
-#         with st.chat_message("assistant"):
-#             chitchat
+import streamlit as st 
+import json
+import google.generativeai as genai
+
+
+GOOGLE_API_KEY = "AIzaSyCUBaL7TdISL7lRuBy19_X0-OsZfgbIgEc"
+genai.configure(api_key=GOOGLE_API_KEY)
+model = genai.GenerativeModel('gemini-pro')
+
+def add_to_json(goal):
+    try:
+        with open("test.json", "r") as file:
+            data = json.load(file)
+    except FileNotFoundError:
+        data = {"goals": []}  # Create the file with an empty 'goals' list if it doesn't exist
+
+    new_item = {"Goal": goal}
+    data["goals"].append(new_item)
+
+    with open("test.json", "w") as file:
+        json.dump(data, file, indent=4)
+        
+
+
+def main():
+    if prompt := st.chat_input("Hi, how can I help you?"):
+        goals_prompt = f"""Act as a personal assistant... {prompt} """  
+        completion = model.generate_content(goals_prompt)
+        add_to_json(prompt)
+
+        with st.chat_message("Assistant"):
+            st.write(completion.text)
+
+        
+
+    # Display JSON Data
+    if st.button("Show JSON Data"):
+        with open("test.json", "r") as file:
+            data = json.load(file)
+        st.json(data)  # Streamlit's way to display JSON
+
+
+if __name__ == "__main__":
+    main()