Upload 3 files

.streamlit/config.toml

@@ -0,0 +1,6 @@
+[theme]
+primaryColor="#F63366"
+backgroundColor="#24243E"
+secondaryBackgroundColor="#0F0C29"
+textColor="#FFFFFF"
+font="sans serif"

app.py

@@ -0,0 +1,91 @@
+import requests
+import streamlit as st
+import json
+from streamlit_chat import message
+
+st.title("GPTBase")
+
+def clear_submit():
+    st.session_state["submit"] = False
+
+def set_api_key(api_key: str):
+    st.session_state["API_KEY"] = api_key
+
+def set_ai_id(ai_id: str):
+    st.session_state["ai_id"] = ai_id
+
+# Sidebar
+index = None
+doc = None
+with st.sidebar:
+    user_secret = st.text_input(
+        "API Key",
+        type="password",
+        placeholder="Paste your API key here (ak-...)",
+        help="You can get your API key from https://mygpt.felo.me/api-keys.",
+        value=st.session_state.get("API_KEY", ""),
+    )
+    if user_secret:
+        set_api_key(user_secret)
+    
+    user_ai_id = st.text_input(
+        "AI Id",
+        placeholder="Paste your AI Id here",
+        value=st.session_state.get("ai_id", ""),
+    )
+    if user_ai_id:
+        set_ai_id(user_ai_id)
+
+    uploaded_file = st.file_uploader(
+        "Upload a pdf, docx, or txt file",
+        type=["pdf", "docx", "txt"],
+        help="Scanned documents are not supported yet!",
+        on_change=clear_submit,
+    )
+    api_key = st.session_state.get("API_KEY")
+    ai_id = st.session_state.get("ai_id")
+    if not api_key or not ai_id:
+        st.warning("Please enter your API key and AI Id to upload a file.")
+
+    if uploaded_file is not None and "API_KEY" in st.session_state and "ai_id" in st.session_state:
+        
+        file = {"file": uploaded_file}
+        url = f'https://mygpt.felo.me/api/v1/ais/{ai_id}/files'
+        headers = {"Authorization": f"Bearer {api_key}"}
+        response = requests.post(url, headers=headers, files=file)
+        result = response.json()
+        if response.status_code != 200:
+            st.text(result['detail'])
+
+tab1, tab2 = st.tabs(["Introduce", "Ask AI"])
+with tab1:
+    st.markdown("### Use steps")
+    st.write('1.Please fill in your API Key and AI Id.')
+    st.write('2.Upload the file you want to get the answer.')
+    st.write('3.You can ask AI with the file.')
+    # st.markdown("""---""")
+    # st.write('If you have any questions or suggestions, please visit: ')
+
+with tab2:
+    if not api_key or not ai_id:
+        st.write('Please fill in your API Key and AI Id.')
+    st.header("Ask AI some questions about the file you uploaded:")
+    
+    def get_text():
+        input_text = st.text_area("You:", on_change=clear_submit)
+        return input_text
+    user_input = get_text()
+    button = st.button("Submit")
+    
+    if button or st.session_state.get("submit"):
+        if api_key and ai_id and user_input:
+            url = f'https://mygpt.felo.me/api/v1/question/{ai_id}'
+            headers = {"Authorization": f"Bearer {api_key}"}
+            data = {"question": user_input}
+            response = requests.post(url, headers=headers, data=json.dumps(data))
+            result = response.json()
+            if response.status_code != 200:
+                st.text(result['detail'])
+            else:
+                message(result['answer'])
+            

requirements.txt

@@ -0,0 +1,2 @@
+streamlit
+streamlit_chat