Duplicate from helliun/personalChatGPT

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: PersonalChatGPT
+emoji: 📈
+colorFrom: yellow
+colorTo: indigo
+sdk: gradio
+sdk_version: 3.19.1
+app_file: app.py
+pinned: false
+duplicated_from: helliun/personalChatGPT
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,66 @@
+import pinecone 
+from sentence_transformers import SentenceTransformer 
+import openai
+import json
+import gradio as gr
+
+pinecone.init(api_key="60422c7f-218d-4343-8158-5b4df9559628", environment="us-east1-gcp")
+index = pinecone.Index("chatgpt-memory")
+
+openai.api_key = "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"
+
+embmodel = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
+
+
+def get_response(input, conversation):
+  user_message = input
+  if conversation!=[]:
+    assistant_message = conversation[-1]["content"]
+  else:
+    assistant_message = ""
+  messages = [{"role": "system", "content": "You are the assistant of 22 year old Henry Leonardi. You have access to memories from previous conversations."}]
+  embed = embmodel.encode([user_message]).tolist()[0]
+  matches = index.query(embed, top_k=5, include_metadata=True)["matches"]
+  try:
+    best = [m["id"] for m in matches][:3]
+    for b in best:
+      messages.extend(eval(b))
+  except:
+    messages = []
+  pair = []
+  d = {"role":"user","content":user_message}
+  conversation.append(d)
+  messages.extend(conversation[-6:])
+  d["content"] = d["content"][:200]
+  pair.append(d)
+  assistant_message = openai.ChatCompletion.create(model="gpt-3.5-turbo",messages=messages)["choices"][0]["message"].to_dict()["content"]  
+  d = {"role":"assistant","content":assistant_message}
+  conversation.append(d)
+  response = d["content"]
+  d["content"] = d["content"][:200]
+  pair.append(d)
+  if pair[0]["content"]!="bye":
+    pair_str = str(json.dumps(pair).encode('utf-8'))[2:-1]
+    index.upsert([(pair_str, embmodel.encode([pair]).tolist()[0])])
+  return response
+
+def conv(input, history=[]):
+    conversation = []
+    for item in history:
+      conversation.append({"role":"user","content":item[0]})
+      conversation.append({"role":"assistant","content":item[1]})
+    output = get_response(input, conversation)
+    history.append((input, output))
+    return history, history
+
+block = gr.Blocks()
+
+with block:
+    gr.Markdown("""<h1><center>Your Personal Assistant</center></h1>""")
+    chatbot = gr.Chatbot()
+    message = gr.Textbox(placeholder="prompt")
+    state = gr.State([])
+    submit = gr.Button("SEND")
+    submit.click(conv, inputs=[message, state], outputs=[chatbot, state])
+
+block.launch(inline = True)

requirements.txt

@@ -0,0 +1,4 @@
+pinecone-client
+sentence-transformers
+openai
+gradio