fix bugs

app.py

@@ -1,14 +1,10 @@
-import os,re
+import os
 import gradio as gr
 import nest_asyncio
 from llama_index.core import StorageContext, load_index_from_storage
 
 from telcom_core import query_rag_qa, query_graph_rag_qa, plot_full_kg, evaluate_llm
 
-import networkx as nx
-from pyvis.network import Network
-from IPython.display import HTML, Markdown, display
-
 
 # Define constants for index paths
 os.environ["OPENAI_API_KEY"] = os.getenv('oai')
@@ -37,8 +33,8 @@ def query_tqa(query, search_level):
     Returns:
         tuple: Responses and references from both Graph-RAG and RAG models.
     """
-    grag_response, grag_reference, grag_reference_text = query_graph_rag_qa(query, search_level)
-    rag_response, rag_reference, rag_reference_text = query_rag_qa(query, search_level)
+    grag_response, grag_reference, grag_reference_text = query_graph_rag_qa(graph_rag_index, query, search_level)
+    rag_response, rag_reference, rag_reference_text = query_rag_qa(rag_index, query, search_level)    
     return grag_response, grag_reference, grag_reference_text, rag_response, rag_reference, rag_reference_text