Spaces:
Running
Running
update
Browse files- src/pages/{5_Ask_Streaming.py β 1_Ask_Streaming.py} +6 -6
- src/pages/{1_Query_And_Answer.py β 2_SQL_Generation.py} +1 -1
- src/pages/{2_Chart_Generation.py β 3_Chart_Generation.py} +0 -0
- src/pages/{3_Metadata.py β 4_Metadata.py} +0 -0
- src/pages/{4_Knowledge_Management.py β 5_Knowledge_Management.py} +0 -0
src/pages/{5_Ask_Streaming.py β 1_Ask_Streaming.py}
RENAMED
@@ -75,9 +75,9 @@ def main():
|
|
75 |
for message in st.session_state.streaming_messages:
|
76 |
with st.chat_message(message["role"]):
|
77 |
if message["role"] == "user":
|
78 |
-
st.
|
79 |
else:
|
80 |
-
st.
|
81 |
if "sql" in message:
|
82 |
with st.expander("= Generated SQL Query", expanded=False):
|
83 |
st.code(format_sql(message["sql"]), language="sql")
|
@@ -148,7 +148,7 @@ def main():
|
|
148 |
delta = event.get("delta", {})
|
149 |
if delta.get("type") == "text_delta":
|
150 |
final_answer += delta.get("text", "")
|
151 |
-
content_container.
|
152 |
|
153 |
elif event.get("type") == "message_stop":
|
154 |
# Extract final data from the event
|
@@ -184,7 +184,7 @@ def main():
|
|
184 |
assistant_message["sql"] = final_sql
|
185 |
|
186 |
st.session_state.streaming_messages.append(assistant_message)
|
187 |
-
content_container.
|
188 |
|
189 |
# Display SQL query if available
|
190 |
if final_sql:
|
@@ -214,7 +214,7 @@ def main():
|
|
214 |
"content": "I couldn't generate an answer for your question. Please try rephrasing it or make sure it's related to your data."
|
215 |
}
|
216 |
st.session_state.streaming_messages.append(assistant_message)
|
217 |
-
content_container.
|
218 |
|
219 |
except Exception as e:
|
220 |
st.error(f"Error during streaming: {str(e)}")
|
@@ -223,7 +223,7 @@ def main():
|
|
223 |
"content": "Sorry, I couldn't process your request. Please check your API credentials and try again."
|
224 |
}
|
225 |
st.session_state.streaming_messages.append(assistant_message)
|
226 |
-
content_container.
|
227 |
|
228 |
# Clear chat button
|
229 |
if st.sidebar.button("π§Ή Clear Chat History"):
|
|
|
75 |
for message in st.session_state.streaming_messages:
|
76 |
with st.chat_message(message["role"]):
|
77 |
if message["role"] == "user":
|
78 |
+
st.text(message["content"])
|
79 |
else:
|
80 |
+
st.text(message["content"])
|
81 |
if "sql" in message:
|
82 |
with st.expander("= Generated SQL Query", expanded=False):
|
83 |
st.code(format_sql(message["sql"]), language="sql")
|
|
|
148 |
delta = event.get("delta", {})
|
149 |
if delta.get("type") == "text_delta":
|
150 |
final_answer += delta.get("text", "")
|
151 |
+
content_container.text(final_answer)
|
152 |
|
153 |
elif event.get("type") == "message_stop":
|
154 |
# Extract final data from the event
|
|
|
184 |
assistant_message["sql"] = final_sql
|
185 |
|
186 |
st.session_state.streaming_messages.append(assistant_message)
|
187 |
+
content_container.text(final_answer)
|
188 |
|
189 |
# Display SQL query if available
|
190 |
if final_sql:
|
|
|
214 |
"content": "I couldn't generate an answer for your question. Please try rephrasing it or make sure it's related to your data."
|
215 |
}
|
216 |
st.session_state.streaming_messages.append(assistant_message)
|
217 |
+
content_container.text(assistant_message["content"])
|
218 |
|
219 |
except Exception as e:
|
220 |
st.error(f"Error during streaming: {str(e)}")
|
|
|
223 |
"content": "Sorry, I couldn't process your request. Please check your API credentials and try again."
|
224 |
}
|
225 |
st.session_state.streaming_messages.append(assistant_message)
|
226 |
+
content_container.text(assistant_message["content"])
|
227 |
|
228 |
# Clear chat button
|
229 |
if st.sidebar.button("π§Ή Clear Chat History"):
|
src/pages/{1_Query_And_Answer.py β 2_SQL_Generation.py}
RENAMED
@@ -6,7 +6,7 @@ from utils import format_sql
|
|
6 |
|
7 |
|
8 |
def main():
|
9 |
-
st.title("π¬ Wren AI Cloud API Demo -
|
10 |
|
11 |
if "api_key" not in st.session_state or "project_id" not in st.session_state:
|
12 |
st.error("Please enter your API Key and Project ID in the sidebar of Home page to get started.")
|
|
|
6 |
|
7 |
|
8 |
def main():
|
9 |
+
st.title("π¬ Wren AI Cloud API Demo - SQL Generation")
|
10 |
|
11 |
if "api_key" not in st.session_state or "project_id" not in st.session_state:
|
12 |
st.error("Please enter your API Key and Project ID in the sidebar of Home page to get started.")
|
src/pages/{2_Chart_Generation.py β 3_Chart_Generation.py}
RENAMED
File without changes
|
src/pages/{3_Metadata.py β 4_Metadata.py}
RENAMED
File without changes
|
src/pages/{4_Knowledge_Management.py β 5_Knowledge_Management.py}
RENAMED
File without changes
|