cyyeh commited on
Commit
99a7fbb
Β·
1 Parent(s): 7df9c64
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.write(message["content"])
79
  else:
80
- st.write(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,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.markdown(final_answer)
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.write(final_answer)
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.write(assistant_message["content"])
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.write(assistant_message["content"])
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 - Query and Answer")
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