app

README.md

@@ -1,6 +1,6 @@
 ---
 title: Chat With Simpsons
-emoji: 📉
+emoji: 🍩
 colorFrom: green
 colorTo: gray
 sdk: streamlit

app.py

@@ -0,0 +1,103 @@
+import streamlit as st
+import json
+import requests
+
+import time
+
+
+API_TOKEN = "api_kqhWITTFmwSnwjsssxKtRxwSDOuPzHphHD"
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+API_URL = "https://api-inference.huggingface.co/models/arampacha/DialoGPT-medium-simpsons"
+
+
+def query(payload):
+    data = json.dumps(payload)
+    response = requests.request("POST", API_URL, headers=headers, data=data)
+    return json.loads(response.content.decode("utf-8"))
+
+def fake_query(payload):
+    user_input = payload["inputs"]["text"]
+    time.sleep(1)
+    res = {
+        "generated_text": user_input[::-1],
+        "conversation":{
+            "past_user_inputs": st.session_state.past_user_inputs + [user_input],
+            "generated_responses": st.session_state.generated_responses + [user_input[::-1]],
+        },
+    }
+    return res
+
+parameters = {
+    "min_length":None,
+    "max_length":50,
+    "top_p":0.92,
+    "repetition_penalty":None,
+}
+
+# options = {}
+
+def on_input():
+    # st.write("Input changed")
+    if st.session_state.count > 0:
+        user_input = st.session_state.user_input
+        st.session_state.full_text += f"_User_  >>> {user_input}\n\n"
+        dialog_output.markdown(st.session_state.full_text)
+        st.session_state.user_input = ""
+
+        payload = {
+            "inputs": {
+                "text": user_input,
+                "past_user_inputs": st.session_state.past_user_inputs,
+                "generated_responses": st.session_state.generated_responses,
+            },
+            "parameters": parameters,
+        }
+        # result = fake_query(payload)
+        result = query(payload)
+        st.session_state.update(result["conversation"])
+        st.session_state.full_text += f'_Chatbot_ > {result["generated_text"]}\n\n'
+    st.session_state.count += 1
+
+
+
+# init session state
+if "past_user_inputs" not in st.session_state:
+    st.session_state["past_user_inputs"] = []
+if "generated_responses" not in st.session_state:
+    st.session_state["generated_responses"] = []
+if "full_text" not in st.session_state:
+    st.session_state["full_text"] = ""
+if "user_input" not in st.session_state:
+    st.session_state["user_input"] = ""
+if "count" not in st.session_state:
+    st.session_state["count"] = 0
+
+# body
+st.title("Chat with Simpsons")
+
+st.image(
+    "https://raw.githubusercontent.com/arampacha/chat-with-simpsons/main/the-simpsons.png",
+    caption="(c) 20th Century Fox Television",
+)
+if st.session_state.count == 0:
+    st.write("Start dialog by inputing some text:")
+
+dialog_output = st.empty()
+
+if st.session_state.count > 0:
+    dialog_output.markdown(st.session_state.full_text)
+
+user_input = st.text_input(
+    "> User: ",
+    # value="Hey Homer! How is it going?",
+    on_change=on_input(),
+    key="user_input",
+)
+
+dialog_text = st.session_state.full_text
+dialog_output.markdown(dialog_text)
+
+def restart():
+    st.session_state.clear()
+    
+st.button("Restart", on_click=st.session_state.clear)