Basic chat interface

app.py

@@ -1,4 +1,85 @@
+import os
 import streamlit as st
+import requests
+from dotenv import load_dotenv
 
-x = st.slider("Select a value")
-st.write(x, "squared is", x * x)
+load_dotenv()
+
+st.set_page_config(page_title="ASI1-mini Chatbot", layout="centered")
+st.title("ASI1-mini LLM Chat")
+st.caption("Powered by [asi1.ai](https://docs.asi1.ai/)")
+
+# Sidebar API Key Input
+st.sidebar.subheader("🔑 API Key")
+asi_api_key = st.sidebar.text_input("ASI1 API Key", type="password")
+st.sidebar.caption(
+    "Get your API key from [asi1.ai](https://asi1.ai/dashboard/api-keys)"
+)
+
+# API Docs Button
+st.sidebar.markdown(
+    """
+    <a href="https://docs.asi1.ai/docs/" target="_blank">
+        <button style="width: 100%; padding: 0.5em; font-size: 16px; background-color: #4CAF50; color: white; border: none; border-radius: 5px;">
+            🔍 Looking for API docs?
+        </button>
+    </a>
+    """,
+    unsafe_allow_html=True,
+)
+
+# Chat history initialization
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
+
+# Display past chat messages
+for msg in st.session_state.chat_history:
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"])
+
+# Chat input box
+user_input = st.chat_input("Ask me anything...")
+
+
+def call_asi1_api(api_key, messages):
+    url = "https://api.asi1.ai/v1/chat/completions"
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json",
+    }
+    payload = {
+        "model": "asi1-mini",
+        "messages": messages,
+        "temperature": 0.7,
+        "fun_mode": False,
+        "web_search": False,
+        "stream": False,
+        "max_tokens": 1024,
+    }
+
+    response = requests.post(url, headers=headers, json=payload)
+    response.raise_for_status()
+    data = response.json()
+    return data.get("choices", [{}])[0].get("message", {}).get("content", "")
+
+
+# When user submits a message
+if user_input:
+    if not asi_api_key:
+        st.error("Please enter your API key in the sidebar.")
+    else:
+        # Add user message to chat history
+        st.session_state.chat_history.append({"role": "user", "content": user_input})
+        with st.chat_message("user"):
+            st.markdown(user_input)
+
+        with st.chat_message("assistant"):
+            with st.spinner("Thinking..."):
+                try:
+                    reply = call_asi1_api(asi_api_key, st.session_state.chat_history)
+                    st.markdown(reply)
+                    st.session_state.chat_history.append(
+                        {"role": "assistant", "content": reply}
+                    )
+                except Exception as e:
+                    st.error(f"Error: {e}")