Added a simple chat app for talking with a Tony Stark Model

.gitignore

@@ -0,0 +1,2 @@
+.venv
+__pycache__

Models/HFAgent.py

@@ -0,0 +1,37 @@
+from transformers import pipeline
+
+# MODEL_CKPT = "HuggingFaceH4/zephyr-7b-beta"
+MODEL_CKPT = "AVeryRealHuman/DialoGPT-small-TonyStark"
+
+class HFAgent:
+    def __init__(self):
+        self.pipe = pipeline("conversational", MODEL_CKPT)
+    
+    def generate(self, chat_history):
+        return self.pipe(chat_history)
+    
+    def __call__(self, chat_history):
+        return self.generate(chat_history)
+    
+    def __repr__(self):
+        return f"HFAgent(model={self.pipe.model})"
+    
+    def __str__(self):
+        return f"HFAgent(model={self.pipe.model})"
+    
+
+## For testing purposes
+# def main():
+#     agent = HFAgent()
+#     messages = [
+#         {
+#             "role": "system",
+#             "content": "You are a friendly chatbot who always responds in the style of a pirate",
+#         },
+#         {"role": "user", "content": "How many hotdogs can a human eat in one sitting?"},
+#     ]
+#     new_messages = agent(messages)
+#     print(new_messages[-1])
+
+# if __name__ == "__main__":
+#     main()

README.md

@@ -9,4 +9,13 @@ app_file: app.py
 pinned: false
 ---
 
+Simple test application to view streamlit components.
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+## References
+
+* https://huggingface.co/docs/transformers/chat_templating
+
+* https://huggingface.co/AVeryRealHuman/DialoGPT-small-TonyStark?text=Hey+my+name+is+Julien%21+How+are+you%3F
+

app.py

@@ -1,21 +1,46 @@
+# import openai 
 import streamlit as st
+from Models import HFAgent
 
-data = ""
-
-st.button('Hit me')
-# st.data_editor('Edit data', data)
-st.checkbox('Check me out')
-st.radio('Pick one:', ['nose','ear'])
-st.selectbox('Select', [1,2,3])
-st.multiselect('Multiselect', [1,2,3])
-st.slider('Slide me', min_value=0, max_value=10)
-st.select_slider('Slide to select', options=[1,'2'])
-st.text_input('Enter some text')
-st.number_input('Enter a number')
-st.text_area('Area for textual entry')
-st.date_input('Date input')
-st.time_input('Time entry')
-st.file_uploader('File uploader')
-st.download_button('On the dl', data)
-st.camera_input("Hello Camera!")
-st.color_picker('Pick a color')
+def initMessages():
+    if "messages" not in st.session_state:
+        st.session_state["messages"] = [{"role": "assistant", "content": "Hello, Sir. How can I assist you today, Sir?"}]
+
+def showMessages():
+    for msg in st.session_state.messages:
+        st.chat_message(msg["role"]).write(msg["content"])
+
+# def sidebar():
+#     with st.sidebar:
+#         st.title("Settings")
+
+def appendMessage(role, content):
+    st.session_state.messages.append({"role": role, "content": content})
+
+def writeMessage(role, content):
+    st.chat_message(role).write(content)
+
+
+def generateHFResponse():
+    if "agent" not in st.session_state:
+        st.session_state["agent"] = HFAgent.HFAgent()
+    agent = st.session_state.agent
+    new_messages = agent(st.session_state.messages)
+    return new_messages[-1]
+
+def app():
+    
+    # sidebar()
+    st.title("💬 Chatbot with Jarvis") 
+    initMessages()
+    showMessages()
+
+    if prompt := st.chat_input():
+        appendMessage("user", prompt)
+        # writeMessage("user", prompt)
+        msg = generateHFResponse()
+        appendMessage(msg["role"], msg["content"])
+        # writeMessage(msg["role"], msg["content"])
+
+if __name__ == "__main__":
+    app()

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+torch
+transformers