vdjvehg

.gitignore

@@ -14,4 +14,6 @@ env/
 .env
 demo.ipynb
 demo.py
-test-1.py
+test-1.py
+train.py
+retrieve.ipynb

app.py

@@ -460,7 +460,7 @@ def handle_message(data):
     context = "No data available"
     session_id = request.sid
     if session_id not in session_store:
-        session_store[session_id] = {'image_data': "", 'message': None, 'image_received': False}
+        session_store[session_id] = {'image_data': b"", 'message': None, 'image_received': False}
 
     if 'message' in data:
         session_store[session_id]['message'] = data['message']

requirements.txt

@@ -1,23 +1,152 @@
-pytorch_lightning
-hydra-core
-lightning
-einops
-pandas
-opencv-python
-timm
-fairscale
-tabulate
-transformers
-gradio
-langchain
-langchain-community
-langchain-groq
-flask
-flask-cors
-flask-socketio
-python-dotenv
-eventlet
-gunicorn
-opencv-python
-spaces
-openai
+aiofiles==23.2.1
+aiohappyeyeballs==2.4.3
+aiohttp==3.10.10
+aiosignal==1.3.1
+annotated-types==0.7.0
+antlr4-python3-runtime==4.9.3
+anyio==4.6.2.post1
+appnope==0.1.4
+asttokens==2.4.1
+async-timeout==4.0.3
+attrs==24.2.0
+bidict==0.23.1
+blinker==1.8.2
+certifi==2024.8.30
+charset-normalizer==3.4.0
+click==8.1.7
+comm==0.2.2
+contourpy==1.3.0
+cycler==0.12.1
+dataclasses-json==0.6.7
+debugpy==1.8.7
+decorator==5.1.1
+distro==1.9.0
+dnspython==2.7.0
+einops==0.8.0
+eventlet==0.37.0
+exceptiongroup==1.2.2
+executing==2.1.0
+fairscale==0.4.13
+fastapi==0.115.3
+ffmpy==0.4.0
+filelock==3.16.1
+Flask==3.0.3
+Flask-Cors==5.0.0
+Flask-SocketIO==5.4.1
+fonttools==4.54.1
+frozenlist==1.5.0
+fsspec==2024.10.0
+gradio==4.44.1
+gradio_client==1.3.0
+greenlet==3.1.1
+groq==0.11.0
+gunicorn==23.0.0
+h11==0.14.0
+httpcore==1.0.6
+httpx==0.27.2
+huggingface-hub==0.26.1
+hydra-core==1.3.2
+idna==3.10
+importlib_metadata==8.5.0
+importlib_resources==6.4.5
+ipykernel==6.29.5
+ipython==8.18.1
+itsdangerous==2.2.0
+jedi==0.19.1
+Jinja2==3.1.4
+jiter==0.6.1
+jsonpatch==1.33
+jsonpointer==3.0.0
+jupyter_client==8.6.3
+jupyter_core==5.7.2
+kiwisolver==1.4.7
+langchain==0.3.4
+langchain-community==0.3.3
+langchain-core==0.3.12
+langchain-groq==0.2.0
+langchain-text-splitters==0.3.0
+langsmith==0.1.137
+lightning==2.4.0
+lightning-utilities==0.11.8
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+marshmallow==3.23.0
+matplotlib==3.9.2
+matplotlib-inline==0.1.7
+mdurl==0.1.2
+mpmath==1.3.0
+multidict==6.1.0
+mypy-extensions==1.0.0
+nest-asyncio==1.6.0
+networkx==3.2.1
+numpy==1.26.4
+omegaconf==2.3.0
+openai==1.52.2
+opencv-python==4.10.0.84
+orjson==3.10.10
+packaging==24.1
+pandas==2.2.3
+parso==0.8.4
+pexpect==4.9.0
+pillow==10.4.0
+platformdirs==4.3.6
+prompt_toolkit==3.0.48
+propcache==0.2.0
+psutil==5.9.8
+ptyprocess==0.7.0
+pure_eval==0.2.3
+pydantic==2.9.2
+pydantic-settings==2.6.0
+pydantic_core==2.23.4
+pydub==0.25.1
+Pygments==2.18.0
+pyparsing==3.2.0
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+python-engineio==4.10.1
+python-multipart==0.0.12
+python-socketio==5.11.4
+pytorch-lightning==2.4.0
+pytz==2024.2
+PyYAML==6.0.2
+pyzmq==26.2.0
+regex==2024.9.11
+requests==2.32.3
+requests-toolbelt==1.0.0
+rich==13.9.3
+ruff==0.7.0
+safetensors==0.4.5
+semantic-version==2.10.0
+shellingham==1.5.4
+simple-websocket==1.1.0
+six==1.16.0
+sniffio==1.3.1
+spaces==0.30.4
+SQLAlchemy==2.0.36
+stack-data==0.6.3
+starlette==0.41.0
+sympy==1.13.1
+tabulate==0.9.0
+tenacity==9.0.0
+timm==1.0.11
+tokenizers==0.20.1
+tomlkit==0.12.0
+torch==2.5.0
+torchmetrics==1.5.1
+torchvision==0.20.0
+tornado==6.4.1
+tqdm==4.66.5
+traitlets==5.14.3
+transformers==4.45.2
+typer==0.12.5
+typing-inspect==0.9.0
+typing_extensions==4.12.2
+tzdata==2024.2
+urllib3==2.2.3
+uvicorn==0.32.0
+wcwidth==0.2.13
+websockets==12.0
+Werkzeug==3.0.4
+wsproto==1.2.0
+yarl==1.16.0
+zipp==3.20.2