change analysis

app.py

@@ -1041,9 +1041,6 @@ from langchain.chains import LLMChain
 from langchain_google_genai import GoogleGenerativeAI, GoogleGenerativeAIEmbeddings
 
 
-
-
-
 class EnhancedLegalSearchSystem:
     def __init__(
         self,
@@ -1895,44 +1892,15 @@ class LegalSearchSystem:
                     }
                 ))
         
-        return sorted(documents, key=lambda x: x.metadata['page_number'])
-
-    def _format_documents(self, results: List[Dict]) -> str:
-        """
-        Format search results as markdown.
-        """
-        markdown = "### Retrieved Documents\n\n"
-        for i, result in enumerate(results, 1):
-            markdown += f"**Document {i}** (Page {result['metadata']['page_number']})\n"
-            markdown += f"```\n{result['metadata']['text']}\n```\n\n"
-        return markdown
+        return
     
 
-async def process_query_async(query: str, search_system: LegalSearchSystem, graph_search_system: EnhancedLegalSearchSystem):
-    """
-    Asynchronous function to process both traditional and graph-based searches
-    """
-    if not query.strip():
-        return "Please enter a query", "", "", "", {}, None, ""
-    
-    # Regular search (synchronous)
-    rag_results = search_system.search(query)
-    
-    try:
-        # Graph search (asynchronous)
-        graph_results = await graph_search_system.process_legal_query(query)
-        graph_documents = graph_results.get('documents', "Error processing graph search")
-        graph_concepts = graph_results.get('related_concepts', {})
-
-        # Prepare graph context for LLM
-        graph_context = graph_search_system._prepare_context(
-            [Document(page_content=d, metadata={}) for d in graph_documents.split("```\n\n```")], # This is a crude split, needs improvement
-            graph_concepts['related_concepts'] if isinstance(graph_concepts, dict) and 'related_concepts' in graph_concepts else []
-        )
-
-    except Exception as e:
-        graph_documents = f"Error processing graph search: {str(e)}"
-        graph_concepts = {}
-        graph_context = ""
-
-    graph_data = graph_search_system.generate_document_graph(query)
+if __name__ == "__main__":
+    graph_search_system = EnhancedLegalSearchSystem(
+        google_api_key="AIzaSyDWHGMd8a70RbL3EBenfUwimcAHjhvgM6M",
+        neo4j_url="neo4j+s://a63462d0.databases.neo4j.io",
+        neo4j_username="neo4j",
+        neo4j_password="nU8Ut5N8k7LmQzNPe7vUbpZxMirK8rHrmLuzPc2G_Zc"    
+    )
+    demo = create_interface(graph_search_system)
+    demo.launch()