wip

.gitignore

@@ -3,5 +3,6 @@ bin/
 obj/
 *.pyproj.user
 .streamlit/secrets.toml
+.env
 
 __pycache__/

app.py

@@ -172,16 +172,17 @@ def run_query(query, report_type, top_k , regions, countries, is_debug, index_na
         #descriptions = "\n".join([f"Description of company \"{res['name']}\":  {res['data']['Summary']}.\n" for res in results[:20] if 'Summary' in res['data']])
         messages = oai.call_assistant(query, engine=openai_model)
         results = st.session_state.db_search_results
-        with content_container:
-            for message in list(messages)[::-1]:
-                with st.chat_message(name = message.role):
-                    st.write(message.content[0].text.value)
-                # if message.role == "assistant":
-                #     with st.chat_message(name = message.role, avatar = st.image("resources/raized_logo.png")):
-                #         st.write(message.content[0].text.value)
-                # else:                        
-                #     with st.chat_message(name = message.role):
-                #         st.write(message.content[0].text.value)
+        if not messages is None:        
+            with content_container:
+                for message in list(messages)[::-1]:
+                    with st.chat_message(name = message.role):
+                        st.write(message.content[0].text.value)
+                    # if message.role == "assistant":
+                    #     with st.chat_message(name = message.role, avatar = st.image("resources/raized_logo.png")):
+                    #         st.write(message.content[0].text.value)
+                    # else:                        
+                    #     with st.chat_message(name = message.role):
+                    #         st.write(message.content[0].text.value)
         # st.session_state.messages.append({"role": "user", "content": query})
         # st.session_state.messages.append({"role": "system", "content": m_text})