Update app.py

app.py

@@ -7,6 +7,9 @@ import g4f
 logging.basicConfig(stream=sys.stdout, level=logging.INFO)
 logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
 
+from llama_index.tools import QueryEngineTool, ToolMetadata
+from llama_index.query_engine import SubQuestionQueryEngine 
+
 from llama_index import VectorStoreIndex, SimpleDirectoryReader, ServiceContext
 from llama_index.llms import HuggingFaceLLM
 from langchain.embeddings import HuggingFaceEmbeddings, HuggingFaceInstructEmbeddings 
@@ -47,11 +50,21 @@ from llama_index import StorageContext, load_index_from_storage
 storage_context = StorageContext.from_defaults(persist_dir="./storage")
 # load index
 index = load_index_from_storage(storage_context, service_context =service_context) 
- 
-async def main(query):
 
-    query_engine = index.as_query_engine()
-    response = query_engine.query(query)
+
+query_engine_tools = [
+    QueryEngineTool(
+        query_engine=query_engine, 
+        metadata=ToolMetadata(name='legal_code_gabon', description='Data on the legal codes of Gabon')
+    )
+]
+
+query_engine = SubQuestionQueryEngine.from_defaults(query_engine_tools=query_engine_tools
+
+async def main(query):
+    
+    #query_engine = index.as_query_engine()
+    response = query_engine.aquery(query)
     print(response)
     return response