Update app.py

app.py

@@ -71,27 +71,27 @@ query_tool = FunctionTool.from_defaults(fn=query_generator)
 llm = OpenAI(model="gpt-3.5-turbo", temperature=0)
 agent = ReActAgent.from_tools([query_tool,rag_tool], llm=llm, verbose=True)
 
-# def implement_query(generated_query):
-#     try:
-#         st.write("Raw Query:", generated_query)
+def implement_query(generated_query):
+    try:
+        st.write("Raw Query:", generated_query)
 
-#         if isinstance(generated_query, str):
-#             generated_query = generated_query.replace("'", '"')
-#             query = json.loads(generated_query)
-#         else:
-#             query = generated_query
+        if isinstance(generated_query, str):
+            generated_query = generated_query.replace("'", '"')
+            query = json.loads(generated_query)
+        else:
+            query = generated_query
 
-#         st.write("Validated Query:", query)
+        st.write("Validated Query:", query)
 
 
-#         response = os_client.search(body=query)
-#         return response
-#     except json.JSONDecodeError as e:
-#         st.error("Error: The generated query is not valid JSON.")
-#         st.write(f"JSONDecodeError Details: {e}")
-#     except Exception as e:
-#         st.error(f"Error executing OpenSearch query: {e}")
-#         st.write(f"Exception Details: {e}")
+    #     response = os_client.search(body=query)
+    #     return response
+    # except json.JSONDecodeError as e:
+    #     st.error("Error: The generated query is not valid JSON.")
+    #     st.write(f"JSONDecodeError Details: {e}")
+    # except Exception as e:
+    #     st.error(f"Error executing OpenSearch query: {e}")
+    #     st.write(f"Exception Details: {e}")
 
 
 st.subheader('OpenSearch Assistant')
@@ -110,9 +110,9 @@ if st.button("Submit"):
                     st.write('Used Tool: ',source.tool_name)
                     if source.tool_name =='query_generator':
                         st.write(source.raw_output.content)
-                        # os_response = implement_query(source.raw_output.content)
-                        # st.subheader('OS Response')
-                        # st.write(os_response)
+                        os_response = implement_query(source.raw_output.content)
+                        st.subheader('OS Response')
+                        st.write(os_response)
                     else:
                         st.write(source.raw_output[0][0][1])
             except Exception as e: