Update Dockerfile

Dockerfile

@@ -29,16 +29,37 @@ RUN pip install -e .
 EXPOSE 7860 8000
 
 # ─── Environment Variable to Pick Framework ───────────────────
-# Default to Streamlit (change to "gradio" or "chainlit" at runtime)
 ENV APP_MODE=streamlit
 
 # ─── Entrypoint ───────────────────────────────────────────────
-CMD if [ "$APP_MODE" = "streamlit" ]; then \
-        streamlit run omniscientframework/app.py --server.port=7860 --server.address=0.0.0.0; \
-    elif [ "$APP_MODE" = "gradio" ]; then \
-        python omniscientframework/gradio_app.py; \
-    elif [ "$APP_MODE" = "chainlit" ]; then \
-        chainlit run omniscientframework/chainlit_app.py --host 0.0.0.0 --port 8000; \
+# Detect whether app.py is at /app/app.py or /app/omniscientframework/app.py
+CMD MODE=$(cat .app_mode 2>/dev/null || echo "$APP_MODE") && \
+    if [ "$MODE" = "streamlit" ]; then \
+        if [ -f "app.py" ]; then \
+            echo "🚀 Launching Streamlit (root-level app.py)"; \
+            streamlit run app.py --server.port=7860 --server.address=0.0.0.0; \
+        elif [ -f "omniscientframework/app.py" ]; then \
+            echo "🚀 Launching Streamlit (package-level app.py)"; \
+            streamlit run omniscientframework/app.py --server.port=7860 --server.address=0.0.0.0; \
+        else \
+            echo "❌ Error: app.py not found."; exit 1; \
+        fi; \
+    elif [ "$MODE" = "gradio" ]; then \
+        if [ -f "gradio_app.py" ]; then \
+            python gradio_app.py; \
+        elif [ -f "omniscientframework/gradio_app.py" ]; then \
+            python omniscientframework/gradio_app.py; \
+        else \
+            echo "❌ Error: gradio_app.py not found."; exit 1; \
+        fi; \
+    elif [ "$MODE" = "chainlit" ]; then \
+        if [ -f "chainlit_app.py" ]; then \
+            chainlit run chainlit_app.py --host 0.0.0.0 --port 8000; \
+        elif [ -f "omniscientframework/chainlit_app.py" ]; then \
+            chainlit run omniscientframework/chainlit_app.py --host 0.0.0.0 --port 8000; \
+        else \
+            echo "❌ Error: chainlit_app.py not found."; exit 1; \
+        fi; \
     else \
-        echo "❌ Unknown APP_MODE: $APP_MODE" && exit 1; \
+        echo "❌ Unknown APP_MODE: $MODE" && exit 1; \
     fi