Fix import structure and package configuration

- Simplified import paths in streamlit_app.py and app.py
- Updated .gitignore to allow __init__.py in models directory
- Fixed Python path in Dockerfile
- Simplified setup.py package configuration

.gitignore

@@ -43,7 +43,8 @@ ENV/
 .env.local
 
 # Models and data
-models/
+models/*
+!models/__init__.py
 data/
 
 # System Files

Dockerfile

@@ -62,7 +62,7 @@ ENV HOST=0.0.0.0 \
     STREAMLIT_SERVER_PORT=8501 \
     STREAMLIT_SERVER_HEADLESS=true \
     STREAMLIT_BROWSER_GATHER_USAGE_STATS=false \
-    PYTHONPATH="/app:${PYTHONPATH}" \
+    PYTHONPATH="/app/src:${PYTHONPATH}" \
     PYTHONUNBUFFERED=1
 
 # Install the package in development mode

app.py

@@ -5,13 +5,13 @@ Main entry point for the Agentic Browser application.
 import os
 import sys
 
-# 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)
+# 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)
 
 # Import and run the Streamlit app
-from src.streamlit_app import main
+from streamlit_app import main
 
 if __name__ == "__main__":
     main()

setup.py

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

src/streamlit_app.py

@@ -4,13 +4,8 @@ import os
 import sys
 from pathlib import Path
 
-# 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)
-
-# Import model manager using absolute import
-from src.models.model_manager import model_manager
+# Import model manager directly
+from models.model_manager import model_manager
 
 # Set page config
 st.set_page_config(