File size: 5,608 Bytes
47402a5 b029384 47402a5 b029384 47402a5 cf56666 47402a5 cf56666 47402a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 |
import os
import pandas as pd
import streamlit as st
from llama_index.experimental.query_engine import PandasQueryEngine
from prompts import new_prompt, instruction_str, context
from note_engine import note_engine
from llama_index.core.tools import QueryEngineTool, ToolMetadata
from llama_index.core.agent import ReActAgent
from llama_index.llms.openai import OpenAI
from data_summary import data_summary_tool
from pdf import bangladesh_engine
from dotenv import load_dotenv
# Load environment variables
load_dotenv()
# File paths
conversation_file = os.path.join("data", "conversation.txt")
summary_file = os.path.join("data", "data_summary.txt")
population_path = os.path.join("data", "Population.csv")
population_df = pd.read_csv(population_path)
# Set up the Streamlit app
st.title("🌎 Population and Bangladesh Data Assistant")
# Sidebar for OpenAI API key
api_key = st.sidebar.text_input("Enter your OpenAI API Key", type="password")
if api_key:
os.environ["OPENAI_API_KEY"] = api_key
# Initialize query engines
population_query_engine = PandasQueryEngine(
df=population_df, verbose=True, instruction_str=instruction_str
)
population_query_engine.update_prompts({"pandas_prompt": new_prompt})
tools = [
QueryEngineTool(
query_engine=population_query_engine,
metadata=ToolMetadata(
name="population_data",
description="Provides information about world population and demographics",
),
),
QueryEngineTool(
query_engine=bangladesh_engine,
metadata=ToolMetadata(
name="bangladesh_data",
description="Provides detailed information about Bangladesh",
),
),
]
llm = OpenAI(model="gpt-3.5-turbo")
agent = ReActAgent.from_tools(tools, llm=llm, verbose=True, context=context)
# Sidebar options
st.sidebar.header("Options")
option = st.sidebar.selectbox("Choose an action:", [
"Ask a Question",
"View Previous Conversations",
"View Data Summary",
"Save a Note"
])
# Conversation management
conversation_active = st.session_state.get('conversation_active', False)
if option == "Ask a Question":
if not conversation_active:
st.session_state.conversation_active = True
st.session_state.conversation_history = []
prompt = st.text_area("Enter your query:", key="user_input")
if st.button("Submit"):
if prompt:
result = agent.query(prompt)
response_text = result.response # Extract just the response text
st.write("Response:", response_text) # Show only the response text
# Save the conversation with a timestamp
timestamp = pd.Timestamp.now().strftime("%Y-%m-%d %H:%M:%S")
st.session_state.conversation_history.append((timestamp, prompt, response_text))
else:
st.error("Please enter a query.")
if st.button("Save this Conversation"):
# Save entire conversation history to the file
with open(conversation_file, "a") as file:
for timestamp, user_prompt, bot_response in st.session_state.conversation_history:
file.write(f"Timestamp: {timestamp}\n")
file.write(f"Prompt: {user_prompt}\n")
file.write(f"Response: {bot_response}\n")
file.write("=" * 40 + "\n")
st.success("Conversation saved.")
if st.button("End Conversation"):
st.session_state.conversation_active = False
st.success("Conversation ended.")
# View previous conversations
elif option == "View Previous Conversations":
if os.path.exists(conversation_file):
try:
with open(conversation_file, "r", encoding="utf-8", errors="replace") as file:
content = file.read()
st.text_area("Previous Conversations", content, height=300)
except Exception as e:
st.error(f"An error occurred while reading the file: {e}")
else:
st.warning("No previous conversations found.")
# View data summary
elif option == "View Data Summary":
if os.path.exists(summary_file):
try:
with open(summary_file, "r", encoding="utf-8", errors="replace") as file:
content = file.read()
st.text_area("Data Summary", content, height=300)
except Exception as e:
st.error(f"An error occurred while reading the file: {e}")
else:
st.warning("No data summary found.")
# Save a note
elif option == "Save a Note":
note = st.text_input("Enter a note to save:")
if st.button("Save Note"):
if note:
# Append note to the conversation file with a timestamp
timestamp = pd.Timestamp.now().strftime("%Y-%m-%d %H:%M:%S")
with open(conversation_file, "a") as file:
file.write(f"Timestamp: {timestamp} (Note)\n")
file.write(f"Note: {note}\n")
file.write("=" * 40 + "\n") # Separator for readability
st.success("Note saved.")
else:
st.error("Please enter a note.")
# Instructions
st.sidebar.subheader("Instructions")
st.sidebar.write(
"1. Enter your OpenAI API Key in the sidebar.\n"
"2. Use the sidebar to choose an action: ask a question, view previous conversations, view the data summary, or save a note.\n"
"3. If you ask a question and click save, the conversation will be saved. If you ask multiple questions and then press save, it will save the whole conversation.\n"
"4. The End Conversation button will simply end the conversation without saving anything.\n"
)
|