Fix import issues and update package structure

- Simplified import statements in streamlit_app.py
- Updated app.py to set up Python path correctly
- Fixed Dockerfile working directory
- Updated setup.py to include all packages
- Added MANIFEST.in for proper package distribution

Dockerfile

@@ -68,6 +68,8 @@ ENV HOST=0.0.0.0 \
 # Install the package in development mode
 WORKDIR /app
 COPY . .
+
+# Install the package in development mode
 RUN pip install -e .
 
 # Install Playwright browsers
@@ -76,6 +78,9 @@ RUN playwright install --with-deps
 # Expose port
 EXPOSE 8501
 
+# Set the working directory to the app root
+WORKDIR /app
+
 # Health check
 HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
   CMD curl --fail http://localhost:8501/_stcore/health || exit 1

MANIFEST.in

@@ -0,0 +1,7 @@
+include README.md
+include requirements.txt
+include LICENSE
+recursive-include src *.py
+recursive-include src *.json
+recursive-include src *.yaml
+recursive-include src *.txt

app.py

@@ -5,13 +5,13 @@ Main entry point for the Agentic Browser application.
 import os
 import sys
 
-# Add the src directory to the Python path
-src_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'src')
-if src_dir not in sys.path:
-    sys.path.insert(0, src_dir)
+# Add the project root to the Python path
+project_root = os.path.dirname(os.path.abspath(__file__))
+if project_root not in sys.path:
+    sys.path.insert(0, project_root)
 
 # Import and run the Streamlit app
-from streamlit_app import main
+from src.streamlit_app import main
 
 if __name__ == "__main__":
     main()

setup.py

@@ -28,8 +28,11 @@ setup(
     url="https://huggingface.co/spaces/anu151105/agentic-browser",
     
     # Package configuration
-    packages=find_packages(where='src'),
-    package_dir={'': 'src'},
+    packages=find_packages(where='src') + find_packages(where='src/models'),
+    package_dir={
+        '': 'src',
+        'models': 'src/models'
+    },
     include_package_data=True,
     package_data={
         '': ['*.json', '*.yaml', '*.txt', '*.md'],

src/streamlit_app.py

@@ -4,17 +4,13 @@ import os
 import sys
 from pathlib import Path
 
-# Add the src directory to the Python path
-src_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-if src_dir not in sys.path:
-    sys.path.insert(0, src_dir)
+# Add the project root to the Python path
+project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+if project_root not in sys.path:
+    sys.path.insert(0, project_root)
 
-try:
-    # Try relative import first
-    from .models.model_manager import model_manager
-except ImportError:
-    # Fall back to absolute import
-    from models.model_manager import model_manager
+# Import model manager using absolute import
+from src.models.model_manager import model_manager
 
 # Set page config
 st.set_page_config(