Spaces:
Build error
Build error
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +17 -17
src/streamlit_app.py
CHANGED
|
@@ -43,7 +43,7 @@ embeddings_model = OpenAIEmbeddings(openai_api_key=API_KEY)
|
|
| 43 |
|
| 44 |
# --- Streamlit Page Setup ---
|
| 45 |
st.set_page_config(page_title="RAG File Chat", layout="centered")
|
| 46 |
-
st.title("
|
| 47 |
|
| 48 |
# --- Session State ---
|
| 49 |
if "uploaded_file" not in st.session_state:
|
|
@@ -72,50 +72,50 @@ def create_agent_and_index(file_content, file_type):
|
|
| 72 |
df = pd.read_csv(io.StringIO(file_content.decode("utf-8")))
|
| 73 |
llm = OpenAI(openai_api_key=API_KEY)
|
| 74 |
st.session_state.agent = create_pandas_dataframe_agent(llm, df, verbose=False)
|
| 75 |
-
st.success("
|
| 76 |
elif file_type == "xlsx":
|
| 77 |
df = pd.read_excel(file_content)
|
| 78 |
llm = OpenAI(openai_api_key=API_KEY)
|
| 79 |
st.session_state.agent = create_pandas_dataframe_agent(llm, df, verbose=False)
|
| 80 |
-
st.success("
|
| 81 |
elif file_type == "json":
|
| 82 |
df = pd.DataFrame(json.loads(file_content.decode("utf-8")))
|
| 83 |
llm = OpenAI(openai_api_key=API_KEY)
|
| 84 |
st.session_state.agent = create_pandas_dataframe_agent(llm, df, verbose=False)
|
| 85 |
-
st.success("
|
| 86 |
elif file_type in ["pdf", "docx"]:
|
| 87 |
text = extract_text_from_file(file_content, file_type)
|
| 88 |
chunks = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0).split_text(text)
|
| 89 |
st.session_state.vectorstore = FAISS.from_texts(chunks, embeddings_model)
|
| 90 |
-
st.success("
|
| 91 |
else:
|
| 92 |
-
st.error("
|
| 93 |
return
|
| 94 |
st.session_state.file_uploaded = True
|
| 95 |
st.session_state.file_type = file_type
|
| 96 |
|
| 97 |
# --- File Upload UI ---
|
| 98 |
-
uploaded = st.file_uploader("
|
| 99 |
if uploaded:
|
| 100 |
st.session_state.uploaded_file = uploaded
|
| 101 |
-
st.info(f"
|
| 102 |
|
| 103 |
-
if st.session_state.uploaded_file and st.button("
|
| 104 |
content = st.session_state.uploaded_file.read()
|
| 105 |
ftype = st.session_state.uploaded_file.name.split(".")[-1].lower()
|
| 106 |
-
with st.spinner("
|
| 107 |
create_agent_and_index(content, ftype)
|
| 108 |
|
| 109 |
# --- Query UI ---
|
| 110 |
if st.session_state.file_uploaded:
|
| 111 |
-
output_format = st.selectbox("
|
| 112 |
-
query = st.text_area("
|
| 113 |
|
| 114 |
if st.button("Submit Query"):
|
| 115 |
if not query.strip():
|
| 116 |
-
st.warning("
|
| 117 |
else:
|
| 118 |
-
with st.spinner("
|
| 119 |
if st.session_state.file_type in ["pdf", "docx"]:
|
| 120 |
qa_chain = RetrievalQA.from_chain_type(
|
| 121 |
llm=OpenAI(openai_api_key=API_KEY),
|
|
@@ -127,7 +127,7 @@ if st.session_state.file_uploaded:
|
|
| 127 |
else:
|
| 128 |
response = st.session_state.agent.run(query)
|
| 129 |
|
| 130 |
-
st.subheader("
|
| 131 |
if output_format == "Plain Text":
|
| 132 |
st.text(response)
|
| 133 |
elif output_format == "Markdown":
|
|
@@ -140,5 +140,5 @@ if st.session_state.file_uploaded:
|
|
| 140 |
df = pd.DataFrame(rows[1:], columns=rows[0])
|
| 141 |
st.dataframe(df)
|
| 142 |
except Exception:
|
| 143 |
-
st.warning("
|
| 144 |
-
st.text(response)
|
|
|
|
| 43 |
|
| 44 |
# --- Streamlit Page Setup ---
|
| 45 |
st.set_page_config(page_title="RAG File Chat", layout="centered")
|
| 46 |
+
st.title("π§ Chat with Your Uploaded File")
|
| 47 |
|
| 48 |
# --- Session State ---
|
| 49 |
if "uploaded_file" not in st.session_state:
|
|
|
|
| 72 |
df = pd.read_csv(io.StringIO(file_content.decode("utf-8")))
|
| 73 |
llm = OpenAI(openai_api_key=API_KEY)
|
| 74 |
st.session_state.agent = create_pandas_dataframe_agent(llm, df, verbose=False)
|
| 75 |
+
st.success("π€ Agent created for CSV.")
|
| 76 |
elif file_type == "xlsx":
|
| 77 |
df = pd.read_excel(file_content)
|
| 78 |
llm = OpenAI(openai_api_key=API_KEY)
|
| 79 |
st.session_state.agent = create_pandas_dataframe_agent(llm, df, verbose=False)
|
| 80 |
+
st.success("π€ Agent created for Excel.")
|
| 81 |
elif file_type == "json":
|
| 82 |
df = pd.DataFrame(json.loads(file_content.decode("utf-8")))
|
| 83 |
llm = OpenAI(openai_api_key=API_KEY)
|
| 84 |
st.session_state.agent = create_pandas_dataframe_agent(llm, df, verbose=False)
|
| 85 |
+
st.success("π€ Agent created for JSON.")
|
| 86 |
elif file_type in ["pdf", "docx"]:
|
| 87 |
text = extract_text_from_file(file_content, file_type)
|
| 88 |
chunks = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0).split_text(text)
|
| 89 |
st.session_state.vectorstore = FAISS.from_texts(chunks, embeddings_model)
|
| 90 |
+
st.success("π Text embedded into FAISS vectorstore.")
|
| 91 |
else:
|
| 92 |
+
st.error("β Unsupported file type.")
|
| 93 |
return
|
| 94 |
st.session_state.file_uploaded = True
|
| 95 |
st.session_state.file_type = file_type
|
| 96 |
|
| 97 |
# --- File Upload UI ---
|
| 98 |
+
uploaded = st.file_uploader("π Browse and select a file", type=["csv", "xlsx", "json", "pdf", "docx"])
|
| 99 |
if uploaded:
|
| 100 |
st.session_state.uploaded_file = uploaded
|
| 101 |
+
st.info(f"β
File selected: `{uploaded.name}` ({uploaded.size / 1024:.1f} KB)")
|
| 102 |
|
| 103 |
+
if st.session_state.uploaded_file and st.button("π€ Upload File"):
|
| 104 |
content = st.session_state.uploaded_file.read()
|
| 105 |
ftype = st.session_state.uploaded_file.name.split(".")[-1].lower()
|
| 106 |
+
with st.spinner("π Processing file..."):
|
| 107 |
create_agent_and_index(content, ftype)
|
| 108 |
|
| 109 |
# --- Query UI ---
|
| 110 |
if st.session_state.file_uploaded:
|
| 111 |
+
output_format = st.selectbox("π Select Output Format", ["Plain Text", "Markdown", "Tabular View"])
|
| 112 |
+
query = st.text_area("π Ask a question about your uploaded file")
|
| 113 |
|
| 114 |
if st.button("Submit Query"):
|
| 115 |
if not query.strip():
|
| 116 |
+
st.warning("β οΈ Please enter a valid question.")
|
| 117 |
else:
|
| 118 |
+
with st.spinner("π‘ Thinking..."):
|
| 119 |
if st.session_state.file_type in ["pdf", "docx"]:
|
| 120 |
qa_chain = RetrievalQA.from_chain_type(
|
| 121 |
llm=OpenAI(openai_api_key=API_KEY),
|
|
|
|
| 127 |
else:
|
| 128 |
response = st.session_state.agent.run(query)
|
| 129 |
|
| 130 |
+
st.subheader("π Answer")
|
| 131 |
if output_format == "Plain Text":
|
| 132 |
st.text(response)
|
| 133 |
elif output_format == "Markdown":
|
|
|
|
| 140 |
df = pd.DataFrame(rows[1:], columns=rows[0])
|
| 141 |
st.dataframe(df)
|
| 142 |
except Exception:
|
| 143 |
+
st.warning("β οΈ Could not render table. Showing raw text.")
|
| 144 |
+
st.text(response))
|