Update app.py

app.py

@@ -3,6 +3,11 @@ from phi.agent import Agent
 from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
 from phi.vectordb.qdrant import Qdrant
 from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChatimport streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
 from phi.model.openai import OpenAIChat
 from phi.embedder.openai import OpenAIEmbedder
 import tempfile
@@ -34,6 +39,7 @@ from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
 from phi.vectordb.qdrant import Qdrant
 from phi.tools.duckduckgo import DuckDuckGo
 from phi.model.openai import OpenAIChat
+    main() 
 from phi.embedder.openai import OpenAIEmbedder
 import tempfile
 import os
@@ -58,12 +64,42 @@ def init_qdrant():
     """Initialize Qdrant vector database"""
     if not st.session_state.qdrant_api_key:
         raise ValueError("Qdrant API key not provided")
-    if not st.session_state.qdrant_url:
-        raise ValueError("Qdrant URL not provided")
-        
-    return Qdrant(          
-        collection="legal_knowledge",
-        url=st.session_state.qdrant_url,
+    if not st.session_state.qdrant_url:import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
+from phi.embedder.openai import OpenAIEmbedder
+import tempfile
+import os
+
+#initializing the session state variables
+def init_session_state():
+    """Initialize session state variables"""
+    if 'openai_api_key' not in st.session_state:
+        st.session_state.openai_api_key = None
+    if 'qdrant_api_key' not in st.session_state:
+        st.session_state.qdrant_api_key = None
+    if 'qdrant_url' not in st.session_state:
+        st.session_state.qdrant_url = None
+    if 'vector_db' not in st.session_state:
+        st.session_state.vector_db = None
+    if 'legal_team' not in st.session_state:
+        st.session_state.legal_team = None
+    if 'knowledge_base' not in st.session_state:
+        st.session_state.knowledge_base = None
+
+def init_qdrant():
+    """Initialize Qdrant vector database"""
+    if not st.session_state.qdrant_api_key:
+        raise ValueError("Qdrant API key not provided")
+    if not st.session_state.qdrant_url:import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
     main() 
         raise ValueError("Qdrant URL not provided")
         
@@ -89,7 +125,43 @@ def process_document(uploaded_file, vector_db: Qdrant):
         with open(temp_file_path, "wb") as f:
             f.write(uploaded_file.getbuffer())
 
-        try:
+        try:import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
+from phi.embedder.openai import OpenAIEmbedder
+import tempfile
+import os
+
+#initializing the session state variables
+def init_session_state():
+    """Initialize session state variables"""
+    if 'openai_api_key' not in st.session_state:
+        st.session_state.openai_api_key = None
+    if 'qdrant_api_key' not in st.session_state:
+        st.session_state.qdrant_api_key = None
+    if 'qdrant_url' not in st.session_state:
+        st.session_state.qdrant_url = None
+    if 'vector_db' not in st.session_state:
+        st.session_state.vector_db = None
+    if 'legal_team' not in st.session_state:
+        st.session_state.legal_team = None
+    if 'knowledge_base' not in st.session_state:
+        st.session_state.knowledge_base = None
+
+def init_qdrant():
+    """Initialize Qdrant vector database"""
+    if not st.session_state.qdrant_api_key:
+        raise ValueError("Qdrant API key not provided")
+    if not st.session_state.qdrant_url:import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
+    main() 
        
             embedder = OpenAIEmbedder(
                 model="text-embedding-3-small",
@@ -99,7 +171,8 @@ def process_document(uploaded_file, vector_db: Qdrant):
             # Creating knowledge base with explicit Qdrant configuration
             knowledge_base = PDFKnowledgeBase(
                 path=temp_dir, 
-                vector_db=vector_db, import streamlit as st
+                vector_db=vector_db, 
+                reader=PDFReader(chunk=True),import streamlit as st
 from phi.agent import Agent
 from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
 from phi.vectordb.qdrant import Qdrant
@@ -129,14 +202,13 @@ def init_qdrant():
     """Initialize Qdrant vector database"""
     if not st.session_state.qdrant_api_key:
         raise ValueError("Qdrant API key not provided")
-    if not st.session_state.qdrant_url:
-        raise ValueError("Qdrant URL not provided")
-        
-    return Qdrant(          
-        collection="legal_knowledge",
-        url=st.session_state.qdrant_url,
+    if not st.session_state.qdrant_url:import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
     main() 
-                reader=PDFReader(chunk=True),
                 embedder=embedder,
                 recreate_vector_db=True  
             )
@@ -153,18 +225,7 @@ def main():
 
     with st.sidebar:
         st.header("🔑 API Configuration")
-   
-        openai_key = st.text_input(
-            "OpenAI API Key",
-            type="password",
-            value=st.session_state.openai_api_key if st.session_state.openai_api_key else "",
-            help="Enter your OpenAI API key"
-        )
-        if openai_key:
-            st.session_state.openai_api_key = openai_key
-
-        qdrant_key = st.text_input(
-            "Qdrant API Key",import streamlit as st
+   import streamlit as st
 from phi.agent import Agent
 from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
 from phi.vectordb.qdrant import Qdrant
@@ -180,7 +241,47 @@ def init_session_state():
     if 'openai_api_key' not in st.session_state:
         st.session_state.openai_api_key = None
     if 'qdrant_api_key' not in st.session_state:
-        st.session_state.qdrant_apiimport streamlit as st
+        st.session_state.qdrant_api_key = None
+    if 'qdrant_url' not in st.session_state:
+        st.session_state.qdrant_url = None
+    if 'vector_db' not in st.session_state:
+        st.session_state.vector_db = None
+    if 'legal_team' not in st.session_state:
+        st.session_state.legal_team = None
+    if 'knowledge_base' not in st.session_state:
+        st.session_state.knowledge_base = None
+
+def init_qdrant():
+    """Initialize Qdrant vector database"""
+    if not st.session_state.qdrant_api_key:
+        raise ValueError("Qdrant API key not provided")
+    if not st.session_state.qdrant_url:import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
+    main() 
+        openai_key = st.text_input(
+            "OpenAI API Key",
+            type="password",
+            value=st.session_state.openai_api_key if st.session_state.openai_api_key else "",
+            help="Enter your OpenAI API key"
+        )
+        if openai_key:
+            st.session_state.openai_api_key = openai_key
+
+        qdrant_key = st.text_input(
+            "Qdrant API Key",
+            type="password",
+            value=st.session_state.qdrant_api_key if st.session_state.qdrant_api_key else "",
+            help="Enter your Qdrant API key"
+        )
+        if qdrant_key:
+            st.session_state.qdrant_api_key = qdrant_key
+
+        qdrant_url = st.text_input(
+            "Qdrant URL",import streamlit as st
 from phi.agent import Agent
 from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
 from phi.vectordb.qdrant import Qdrant
@@ -210,42 +311,13 @@ def init_qdrant():
     """Initialize Qdrant vector database"""
     if not st.session_state.qdrant_api_key:
         raise ValueError("Qdrant API key not provided")
-    if not st.session_state.qdrant_url:
-        raise ValueError("Qdrant URL not provided")
-        
-    return Qdrant(          
-        collection="legal_knowledge",
-        url=st.session_state.qdrant_url,
-    main() _key = None
-    if 'qdrant_url' not in st.session_state:
-        st.session_state.qdrant_url = None
-    if 'vector_db' not in st.session_state:
-        st.session_state.vector_db = None
-    if 'legal_team' not in st.session_state:
-        st.session_state.legal_team = None
-    if 'knowledge_base' not in st.session_state:
-        st.session_state.knowledge_base = None
-
-def init_qdrant():
-    """Initialize Qdrant vector database"""
-    if not st.session_state.qdrant_api_key:
-        raise ValueError("Qdrant API key not provided")
-    if not st.session_state.qdrant_url:
-        raise ValueError("Qdrant URL not provided")
-        
-    return Qdrant(          
-        collection="legal_knowledge",
-        url=st.session_state.qdrant_url,
+    if not st.session_state.qdrant_url:import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
     main() 
-            type="password",
-            value=st.session_state.qdrant_api_key if st.session_state.qdrant_api_key else "",
-            help="Enter your Qdrant API key"
-        )
-        if qdrant_key:
-            st.session_state.qdrant_api_key = qdrant_key
-
-        qdrant_url = st.text_input(
-            "Qdrant URL",
             value=st.session_state.qdrant_url if st.session_state.qdrant_url else "https://f499085c-b4bf-4bda-a9a5-227f62a9ca20.us-west-2-0.aws.cloud.qdrant.io:6333",
             help="Enter your Qdrant instance URL"
         )
@@ -259,7 +331,10 @@ def init_qdrant():
                     st.success("Successfully connected to Qdrant!")
             except Exception as e:
                 st.error(f"Failed to connect to Qdrant: {str(e)}")
-import streamlit as st
+
+        st.divider()
+
+        if all([st.session_state.openai_api_key, st.session_state.vector_db]):import streamlit as st
 from phi.agent import Agent
 from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
 from phi.vectordb.qdrant import Qdrant
@@ -289,16 +364,13 @@ def init_qdrant():
     """Initialize Qdrant vector database"""
     if not st.session_state.qdrant_api_key:
         raise ValueError("Qdrant API key not provided")
-    if not st.session_state.qdrant_url:
-        raise ValueError("Qdrant URL not provided")
-        
-    return Qdrant(          
-        collection="legal_knowledge",
-        url=st.session_state.qdrant_url,
+    if not st.session_state.qdrant_url:import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
     main() 
-        st.divider()
-
-        if all([st.session_state.openai_api_key, st.session_state.vector_db]):
             st.header("📄 Document Upload")
             uploaded_file = st.file_uploader("Upload Legal Document", type=['pdf'])
             
@@ -317,7 +389,43 @@ def init_qdrant():
                             knowledge=st.session_state.knowledge_base,
                             search_knowledge=True,
                             instructions=[
-                                "Find and cite relevant legal cases and precedents",
+                                "Find and cite relevant legal cases and precedents",import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
+from phi.embedder.openai import OpenAIEmbedder
+import tempfile
+import os
+
+#initializing the session state variables
+def init_session_state():
+    """Initialize session state variables"""
+    if 'openai_api_key' not in st.session_state:
+        st.session_state.openai_api_key = None
+    if 'qdrant_api_key' not in st.session_state:
+        st.session_state.qdrant_api_key = None
+    if 'qdrant_url' not in st.session_state:
+        st.session_state.qdrant_url = None
+    if 'vector_db' not in st.session_state:
+        st.session_state.vector_db = None
+    if 'legal_team' not in st.session_state:
+        st.session_state.legal_team = None
+    if 'knowledge_base' not in st.session_state:
+        st.session_state.knowledge_base = None
+
+def init_qdrant():
+    """Initialize Qdrant vector database"""
+    if not st.session_state.qdrant_api_key:
+        raise ValueError("Qdrant API key not provided")
+    if not st.session_state.qdrant_url:import streamlit as st
+from phi.agent import Agent
+from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
+from phi.vectordb.qdrant import Qdrant
+from phi.tools.duckduckgo import DuckDuckGo
+from phi.model.openai import OpenAIChat
+    main() 
                                 "Provide detailed research summaries with sources",
                                 "Reference specific sections from the uploaded document",
                                 "Always search the knowledge base for relevant information"
@@ -390,43 +498,7 @@ def init_qdrant():
                     "Custom Query"
                 ]
             )
-        else:import streamlit as st
-from phi.agent import Agent
-from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
-from phi.vectordb.qdrant import Qdrant
-from phi.tools.duckduckgo import DuckDuckGo
-from phi.model.openai import OpenAIChat
-from phi.embedder.openai import OpenAIEmbedder
-import tempfile
-import os
-
-#initializing the session state variables
-def init_session_state():
-    """Initialize session state variables"""
-    if 'openai_api_key' not in st.session_state:
-        st.session_state.openai_api_key = None
-    if 'qdrant_api_key' not in st.session_state:
-        st.session_state.qdrant_api_key = None
-    if 'qdrant_url' not in st.session_state:
-        st.session_state.qdrant_url = None
-    if 'vector_db' not in st.session_state:
-        st.session_state.vector_db = None
-    if 'legal_team' not in st.session_state:
-        st.session_state.legal_team = None
-    if 'knowledge_base' not in st.session_state:
-        st.session_state.knowledge_base = None
-
-def init_qdrant():
-    """Initialize Qdrant vector database"""
-    if not st.session_state.qdrant_api_key:
-        raise ValueError("Qdrant API key not provided")
-    if not st.session_state.qdrant_url:
-        raise ValueError("Qdrant URL not provided")
-        
-    return Qdrant(          
-        collection="legal_knowledge",
-        url=st.session_state.qdrant_url,
-    main() 
+        else:
             st.warning("Please configure all API credentials to proceed")
 
     # Main content area
@@ -435,43 +507,7 @@ def init_qdrant():
     elif not uploaded_file:
         st.info("👈 Please upload a legal document to begin analysis")
     elif st.session_state.legal_team:
-        # Create a dictionary for import streamlit as st
-from phi.agent import Agent
-from phi.knowledge.pdf import PDFKnowledgeBase, PDFReader
-from phi.vectordb.qdrant import Qdrant
-from phi.tools.duckduckgo import DuckDuckGo
-from phi.model.openai import OpenAIChat
-from phi.embedder.openai import OpenAIEmbedder
-import tempfile
-import os
-
-#initializing the session state variables
-def init_session_state():
-    """Initialize session state variables"""
-    if 'openai_api_key' not in st.session_state:
-        st.session_state.openai_api_key = None
-    if 'qdrant_api_key' not in st.session_state:
-        st.session_state.qdrant_api_key = None
-    if 'qdrant_url' not in st.session_state:
-        st.session_state.qdrant_url = None
-    if 'vector_db' not in st.session_state:
-        st.session_state.vector_db = None
-    if 'legal_team' not in st.session_state:
-        st.session_state.legal_team = None
-    if 'knowledge_base' not in st.session_state:
-        st.session_state.knowledge_base = None
-
-def init_qdrant():
-    """Initialize Qdrant vector database"""
-    if not st.session_state.qdrant_api_key:
-        raise ValueError("Qdrant API key not provided")
-    if not st.session_state.qdrant_url:
-        raise ValueError("Qdrant URL not provided")
-        
-    return Qdrant(          
-        collection="legal_knowledge",
-        url=st.session_state.qdrant_url,
-    main() analysis type icons
+        # Create a dictionary for analysis type icons
         analysis_icons = {
             "Contract Review": "📑",
             "Legal Research": "🔍",