Initial commit for Hugging Face Space deployment

app.py

@@ -8,11 +8,17 @@ import os
 import sys
 
 # Make sure the app directory is in the path
+# Add the current directory to the path so that 'app' is recognized as a package
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+# Also add the parent directory to path to ensure imports work properly
+sys.path.append(os.path.abspath('.'))
 
 # Create necessary directories
 os.makedirs('data/documents', exist_ok=True)
 os.makedirs('data/vector_db', exist_ok=True)
 
+# Set environment variable for Python path
+os.environ['PYTHONPATH'] = os.path.abspath('.')
+
 # Run the Streamlit app with specific port to match huggingface-space.yml
 subprocess.run(["streamlit", "run", "app/ui/streamlit_app.py", "--server.port=7860", "--server.address=0.0.0.0"]) 

app/__init__.py

@@ -0,0 +1 @@
+# This file marks 'app' as a Python package 

app/core/__init__.py

@@ -0,0 +1 @@
+# This file marks 'app.core' as a Python package 

app/ui/__init__.py

@@ -0,0 +1 @@
+# This file marks 'app.ui' as a Python package 

app/ui/streamlit_app.py

@@ -7,10 +7,20 @@ from typing import List, Dict, Any
 
 # Add project root to path for imports
 sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
-from app.core.agent import AssistantAgent
-from app.core.ingestion import DocumentProcessor
-from app.utils.helpers import get_document_path, format_sources, save_conversation
-from app.config import LLM_MODEL, EMBEDDING_MODEL
+
+# Use relative imports when running as part of the app package
+try:
+    from app.core.agent import AssistantAgent
+    from app.core.ingestion import DocumentProcessor
+    from app.utils.helpers import get_document_path, format_sources, save_conversation
+    from app.config import LLM_MODEL, EMBEDDING_MODEL
+except ImportError:
+    # Fallback to direct imports if app is not recognized as a package
+    sys.path.append(os.path.abspath('.'))
+    from app.core.agent import AssistantAgent
+    from app.core.ingestion import DocumentProcessor
+    from app.utils.helpers import get_document_path, format_sources, save_conversation
+    from app.config import LLM_MODEL, EMBEDDING_MODEL
 
 # Set page config
 st.set_page_config(

app/utils/__init__.py

@@ -0,0 +1 @@
+# This file marks 'app.utils' as a Python package