test

.devcontainer/devcontainer.json

@@ -42,5 +42,9 @@
         // "ghcr.io/devcontainers-contrib/features/coverage-py:2": {},
         // "git": "latest",
         // "github-cli": "latest"
-    }
+    },
+    // add environment variables
+    "containerEnv": {
+      "OPENAI_API_KEY": "${localEnv:OPENAI_API_KEY}"
+  }
 }

streamlit_agent/Untitled-1.ipynb

@@ -0,0 +1,54 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "OperationalError",
+     "evalue": "unable to open database file",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[1;31mOperationalError\u001b[0m                          Traceback (most recent call last)",
+      "Cell \u001b[1;32mIn[2], line 4\u001b[0m\n\u001b[0;32m      1\u001b[0m \u001b[39mimport\u001b[39;00m \u001b[39mpandas\u001b[39;00m \u001b[39mas\u001b[39;00m \u001b[39mpd\u001b[39;00m\n\u001b[0;32m      2\u001b[0m \u001b[39mimport\u001b[39;00m \u001b[39msqlite3\u001b[39;00m\n\u001b[1;32m----> 4\u001b[0m conn \u001b[39m=\u001b[39m sqlite3\u001b[39m.\u001b[39;49mconnect(\u001b[39m'\u001b[39;49m\u001b[39mstreamlit_agent\u001b[39;49m\u001b[39m\\\u001b[39;49m\u001b[39msitios2.sqlite\u001b[39;49m\u001b[39m'\u001b[39;49m)\n\u001b[0;32m      5\u001b[0m df \u001b[39m=\u001b[39m pd\u001b[39m.\u001b[39mread_sql_query(\u001b[39m\"\u001b[39m\u001b[39mSELECT * FROM sitiospilotojson\u001b[39m\u001b[39m\"\u001b[39m, conn)\n\u001b[0;32m      6\u001b[0m conn\u001b[39m.\u001b[39mclose()\n",
+      "\u001b[1;31mOperationalError\u001b[0m: unable to open database file"
+     ]
+    }
+   ],
+   "source": [
+    "import pandas as pd\n",
+    "import sqlite3\n",
+    "\n",
+    "conn = sqlite3.connect('streamlit_agent\\sitios2.sqlite')\n",
+    "df = pd.read_sql_query(\"SELECT * FROM sitiospilotojson\", conn)\n",
+    "conn.close()\n",
+    "\n",
+    "df"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.4"
+  },
+  "orig_nbformat": 4
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

streamlit_agent/mrkl_demo.py

@@ -12,26 +12,28 @@ from langchain.utilities import DuckDuckGoSearchAPIWrapper
 
 from streamlit_agent.callbacks.capturing_callback_handler import playback_callbacks
 from streamlit_agent.clear_results import with_clear_container
+import os
 
-DB_PATH = (Path(__file__).parent / "Chinook.db").absolute()
+user_openai_api_key = os.environ.get('OPENAI_API_KEY')
+
+# DB_PATH = (Path(__file__).parent / "Chinook.db").absolute()
+DB_PATH = (Path(__file__).parent / "sitios2.sqlite").absolute()
 
 SAVED_SESSIONS = {
-    "Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?": "leo.pickle",
-    "What is the full name of the artist who recently released an album called "
-    "'The Storm Before the Calm' and are they in the FooBar database? If so, what albums of theirs "
-    "are in the FooBar database?": "alanis.pickle",
+    "Que provincias están representadas en los sitios pilotos?": "alanis.pickle",
+    "Cual es la superficie total de sitios piloto en Buenos Aires?": "alanis.pickle",
 }
 
 st.set_page_config(
     page_title="MRKL", page_icon="🦜", layout="wide", initial_sidebar_state="collapsed"
 )
 
-"# 🦜🔗 MRKL"
+"# Tabla Sitios Piloto"
 
 # Setup credentials in Streamlit
-user_openai_api_key = st.sidebar.text_input(
-    "OpenAI API Key", type="password", help="Set this to run your own custom questions."
-)
+# user_openai_api_key = st.sidebar.text_input(
+#     "OpenAI API Key", type="password", help="Set this to run your own custom questions."
+# )
 
 if user_openai_api_key:
     openai_api_key = user_openai_api_key
@@ -58,9 +60,9 @@ tools = [
         description="useful for when you need to answer questions about math",
     ),
     Tool(
-        name="FooBar DB",
+        name="sitios piloto DB",
         func=db_chain.run,
-        description="useful for when you need to answer questions about FooBar. Input should be in the form of a question containing full context",
+        description="useful for when you need to answer questions about sitios piloto. Input should be in the form of a question containing full context",
     ),
 ]
 
@@ -69,6 +71,18 @@ mrkl = initialize_agent(
     tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
 )
 
+# read sitios2.sqlite and convert it to dataframe
+
+import pandas as pd
+import sqlite3
+
+conn = sqlite3.connect(DB_PATH)
+df = pd.read_sql_query("SELECT spoat, ecorregion, sag, provincia, supha, area, departamento FROM sitiospilotojson", conn)
+conn.close()
+
+st.write(df)
+
+
 with st.form(key="form"):
     if not enable_custom:
         "Ask one of the sample questions, or enter your API Key in the sidebar to ask your own custom questions."
@@ -77,7 +91,7 @@ with st.form(key="form"):
 
     if enable_custom:
         user_input = st.text_input("Or, ask your own question")
-    if not mrkl_input:
+    if not user_input:
         user_input = prefilled
     submit_clicked = st.form_submit_button("Submit Question")
 
@@ -91,12 +105,12 @@ if with_clear_container(submit_clicked):
 
     # If we've saved this question, play it back instead of actually running LangChain
     # (so that we don't exhaust our API calls unnecessarily)
-    if user_input in SAVED_SESSIONS:
-        session_name = SAVED_SESSIONS[user_input]
-        session_path = Path(__file__).parent / "runs" / session_name
-        print(f"Playing saved session: {session_path}")
-        answer = playback_callbacks([st_callback], str(session_path), max_pause_time=2)
-    else:
-        answer = mrkl.run(user_input, callbacks=[st_callback])
-
+    # if user_input in SAVED_SESSIONS:
+    #     session_name = SAVED_SESSIONS[user_input]
+    #     session_path = Path(__file__).parent / "runs" / session_name
+    #     print(f"Playing saved session: {session_path}")
+    #     answer = playback_callbacks([st_callback], str(session_path), max_pause_time=2)
+    # else:
+    #     answer = mrkl.run(user_input, callbacks=[st_callback])
+    answer = mrkl.run(user_input, callbacks=[st_callback])
     answer_container.write(answer)