Update app.py

app.py

@@ -1,22 +1,52 @@
+import streamlit as st
 import requests
 
+# Set up Streamlit page configuration
+st.set_page_config(page_title="DeepSeek Chatbot", page_icon="🤖", layout="wide")
+
+# API setup
 url = "https://api.hyperbolic.xyz/v1/chat/completions"
 headers = {
     "Content-Type": "application/json",
     "Authorization": "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJtaXRyYWxlc3RhcmlwZXJzYWRhQGdtYWlsLmNvbSIsImlhdCI6MTczNjUwMzQxMX0.yuIoZsH1jouAlixx_h_eQ-bltZ1sg4alrJHMHr1axvA"
 }
-data = {
-    "messages": [
-        {
-            "role": "user",
-            "content": "What can I do in SF?"
-        }
-    ],
-    "model": "deepseek-ai/DeepSeek-V3",
-    "max_tokens": 512,
-    "temperature": 0.1,
-    "top_p": 0.9
-}
-  
-response = requests.post(url, headers=headers, json=data)
-print(response.json())
+
+# Chat history container
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+
+# Function to send message and get response
+def get_response(user_input):
+    data = {
+        "messages": [{"role": "user", "content": user_input}],
+        "model": "deepseek-ai/DeepSeek-V3",
+        "max_tokens": 512,
+        "temperature": 0.1,
+        "top_p": 0.9
+    }
+    
+    response = requests.post(url, headers=headers, json=data)
+    return response.json()
+
+# Streamlit chat UI
+st.title("DeepSeek AI Chatbot")
+
+# Display the chat history
+for message in st.session_state.messages:
+    if message["role"] == "user":
+        st.chat_message("user").markdown(message["content"])
+    else:
+        st.chat_message("assistant").markdown(message["content"])
+
+# Accept user input
+user_input = st.text_input("You: ", "")
+
+# Handle user input and update the chat
+if user_input:
+    st.session_state.messages.append({"role": "user", "content": user_input})
+    response = get_response(user_input)
+    
+    # Assuming the response is in the 'choices' field of the API response
+    bot_response = response.get('choices', [{}])[0].get('message', {}).get('content', 'Sorry, I did not understand that.')
+    st.session_state.messages.append({"role": "assistant", "content": bot_response})
+    st.experimental_rerun()